fix mypy typing on optional times

OpenFreeEnergy · Jun 30, 2023 · 980b812 · 980b812
1 parent daf6d84
commit 980b812
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gufe/protocols/protocolunit.py b/gufe/protocols/protocolunit.py
@@ -142,12 +142,12 @@ def ok() -> bool:
         return True
 
     @property
-    def start_time(self) -> datetime.datetime:
+    def start_time(self) -> Optional[datetime.datetime]:
         """The time execution of this Unit began"""
         return self._start_time
 
     @property
-    def end_time(self) -> datetime.datetime:
+    def end_time(self) -> Optional[datetime.datetime]:
         """The time at which execution of this Unit finished"""
         return self._end_time