neuralcompression/models/scale_hyperprior.py (2 lines): - line 219: TODO: Add proper types for bottleneck component args instead of using - line 402: # TODO: Switch to named tuple