From dd9a286ad67615efaa36ddd29e2cf533e4a2b8c1 Mon Sep 17 00:00:00 2001
From: Stefano Borini <sborini@enthought.com>
Date: Tue, 24 Apr 2018 14:32:12 +0100
Subject: [PATCH] Removed __traits_version__ from model getstate
Removes the __traits_version__ when dumping the state with __getstate__()
---
force_bdss/io/workflow_writer.py | 14 +++++++++++++-
1 file changed, 13 insertions(+), 1 deletion(-)
diff --git a/force_bdss/io/workflow_writer.py b/force_bdss/io/workflow_writer.py
index 3a2eee7..b39c855 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__()
--
GitLab