diff --git a/force_bdss/io/workflow_writer.py b/force_bdss/io/workflow_writer.py index 3a2eee725c519502759f240a7e29c554f1e0f458..b39c8550141c69322f23cd9f3e8e79bddfad24b2 100644 --- a/force_bdss/io/workflow_writer.py +++ b/force_bdss/io/workflow_writer.py @@ -49,10 +49,16 @@ class WorkflowWriter(HasStrictTraits): parameters_data = [] for param in data["model_data"]["parameters"]: + state = param.__getstate__() + try: + state.pop("__traits_version__") + except KeyError: + pass + parameters_data.append( { "id": param.factory.id, - "model_data": param.__getstate__() + "model_data": state } ) @@ -63,6 +69,12 @@ class WorkflowWriter(HasStrictTraits): """ Extracts the data from an external model and returns its dictionary """ + state = model.__getstate__() + try: + state.pop("__traits_version__") + except KeyError: + pass + return { "id": model.factory.id, "model_data": model.__getstate__()