def __post_init__()

in neuron_explainer/activations/derived_scalars/activations_and_metadata.py [0:0]

• 6 lines of code
• 5 McCabe index (conditional complexity)

    def __post_init__(self) -> None:
        # assert that all the devices are the same or None
        device = self.device
        if len(self.activations_by_layer_index) > 0:
            for activations in self.activations_by_layer_index.values():
                if activations.device is not None:
                    assert activations.device == device