diff --git a/src/jobflow/core/job.py b/src/jobflow/core/job.py index 6d036a2b..378f43d2 100644 --- a/src/jobflow/core/job.py +++ b/src/jobflow/core/job.py @@ -623,9 +623,7 @@ def run(self, store: jobflow.JobStore) -> Response: pass_manager_config(response.replace, passed_config) try: - output = jsanitize( - response.output, strict=True, enum_values=True, allow_bson=True - ) + output = jsanitize(response.output, strict=True, allow_bson=True) except AttributeError as err: raise RuntimeError( "Job output contained an object that is not MSONable and therefore " @@ -1106,7 +1104,7 @@ def as_dict(self) -> dict: # fireworks can't serialize functions and classes, so explicitly serialize to # the job recursively using monty to avoid issues - return jsanitize(d, strict=True, enum_values=True, allow_bson=True) + return jsanitize(d, strict=True, allow_bson=True) def __setattr__(self, key, value): """Handle setting attributes. Implements a special case for job name.""" diff --git a/src/jobflow/core/reference.py b/src/jobflow/core/reference.py index 10d774b3..07e1b1c9 100644 --- a/src/jobflow/core/reference.py +++ b/src/jobflow/core/reference.py @@ -394,7 +394,7 @@ def find_and_get_references(arg: Any) -> tuple[OutputReference, ...]: # argument is a primitive, we won't find a reference here return () - arg = jsanitize(arg, strict=True, enum_values=True, allow_bson=True) + arg = jsanitize(arg, strict=True, allow_bson=True) # recursively find any reference classes locations = find_key_value(arg, "@class", "OutputReference") @@ -458,7 +458,7 @@ def find_and_resolve_references( return arg # serialize the argument to a dictionary - encoded_arg = jsanitize(arg, strict=True, enum_values=True, allow_bson=True) + encoded_arg = jsanitize(arg, strict=True, allow_bson=True) # recursively find any reference classes locations = find_key_value(encoded_arg, "@class", "OutputReference")