added type annotation to OpacityState getitem

tardis-sn · Jul 31, 2024 · 401c6c5 · 401c6c5
1 parent 43a9762
commit 401c6c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tardis/opacities/opacity_state.py b/tardis/opacities/opacity_state.py
@@ -106,7 +106,7 @@ def __init__(
         self.photo_ion_activation_idx = photo_ion_activation_idx
         self.k_packet_idx = k_packet_idx
 
-    def __getitem__(self, i):
+    def __getitem__(self, i: slice):
         """Get a shell or slice of shells of the attributes of the opacity state
 
         Args: