Extended default values to ProtocolUnitFailure

OpenFreeEnergy · Jun 30, 2023 · daf6d84 · daf6d84
1 parent 4995a91
commit daf6d84
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/gufe/protocols/protocolunit.py b/gufe/protocols/protocolunit.py
@@ -155,8 +155,19 @@ def end_time(self) -> datetime.datetime:
 class ProtocolUnitFailure(ProtocolUnitResult):
     """Failed result for a single `ProtocolUnit` execution."""
 
-    def __init__(self, *, name=None, source_key, inputs, outputs, _key=None, exception, traceback,
-                 start_time: datetime.datetime, end_time: datetime.datetime):
+    def __init__(
+            self,
+            *,
+            name=None,
+            source_key,
+            inputs,
+            outputs,
+            _key=None,
+            exception,
+            traceback,
+            start_time: Optional[datetime.datetime] = None,
+            end_time: Optional[datetime.datetime] = None,
+        ):
         """
         Parameters
         ----------