diff --git a/force_bdss/io/workflow_writer.py b/force_bdss/io/workflow_writer.py index 66379aea9e1970baa9c4ff2c9301b81608462b4c..bb8039e66c6a523281ca35e0c1816cb19564b647 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__()