From 290a8013b8f919a65a612c12908791d9473d13ba Mon Sep 17 00:00:00 2001
From: Nikolai <osterei33@gmx.de>
Date: Thu, 16 Aug 2018 17:42:49 +0200
Subject: [PATCH] python3 compatibility
---
toolkit.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index 86f3f23..67c1764 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -81,7 +81,7 @@ def load_from_dir(path):
project_type = info["project_type"]
if project_type == "ClassificationProjectRNN":
return ClassificationProjectRNN(path)
- except KeyError, IOError:
+ except (KeyError, IOError):
pass
return ClassificationProject(path)
@@ -705,7 +705,7 @@ class ClassificationProject(object):
of.write(self._model.to_json())
# plot model
- with open(os.path.join(self.project_dir, "model.svg"), "w") as of:
+ with open(os.path.join(self.project_dir, "model.svg"), "wb") as of:
of.write(model_to_dot(self._model).create("dot", format="svg"))
--
GitLab