Make behavior day optional

AllenInstitute · Dec 6, 2023 · a15c39d · a15c39d
1 parent 9f395fb
commit a15c39d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/npc_lims/status/tracked_sessions.py b/src/npc_lims/status/tracked_sessions.py
@@ -78,10 +78,10 @@ def cloud_path(self) -> upath.UPath | None:
         return None
 
     @functools.cached_property
-    def behavior_day(self) -> int:
+    def behavior_day(self) -> int | None:
         if self.is_templeton:
             raise AttributeError("`behavior_day` is not defined for Templeton sessions")
-        return self.training_info["ID"]  # row of training spreadsheet
+        return self.training_info.get("ID")  # row of training spreadsheet
 
     @functools.cached_property
     def is_uploaded(self) -> bool: