def get_parameterized_dimension_sizes(self) -> dict[Dimension, int]:
        """A dictionary specifying the size of the parameterized dimensions; for convenient use with ScalarDerivers"""
        return {
            Dimension.AUTOENCODER_LATENTS: self.num_autoencoder_directions,
        }
A dictionary specifying the size of the parameterized dimensions; for convenient use with ScalarDerivers
neuron_explainer/models/autoencoder_context.py