Lint.

AllenNeuralDynamics · Nov 14, 2024 · 59158c1 · 59158c1
1 parent ffce70a
commit 59158c1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/aind_metadata_mapper/stimulus/camstim.py b/src/aind_metadata_mapper/stimulus/camstim.py
@@ -86,17 +86,17 @@ def __init__(
 
     def _get_session_type(self) -> str:
         """Determine the session type from the pickle data
-        
+
         Returns
         -------
         str
             session type
         """
         if self.behavior:
-            return self.pkl_data['items']["behavior"]['params']['stage']
+            return self.pkl_data["items"]["behavior"]["params"]["stage"]
         else:
-            return self.pkl_data['items']["foraging"]["params"]["stage"]
-        
+            return self.pkl_data["items"]["foraging"]["params"]["stage"]
+
     def _is_behavior(self) -> bool:
         """Check if the session has behavior data"""
         if self.pkl_data.get("items", {}).get("behavior", None):