Skip to content
Snippets Groups Projects
Commit 290a8013 authored by Nikolai's avatar Nikolai
Browse files

python3 compatibility

parent ae824de0
No related branches found
No related tags found
No related merge requests found
...@@ -81,7 +81,7 @@ def load_from_dir(path): ...@@ -81,7 +81,7 @@ def load_from_dir(path):
project_type = info["project_type"] project_type = info["project_type"]
if project_type == "ClassificationProjectRNN": if project_type == "ClassificationProjectRNN":
return ClassificationProjectRNN(path) return ClassificationProjectRNN(path)
except KeyError, IOError: except (KeyError, IOError):
pass pass
return ClassificationProject(path) return ClassificationProject(path)
...@@ -705,7 +705,7 @@ class ClassificationProject(object): ...@@ -705,7 +705,7 @@ class ClassificationProject(object):
of.write(self._model.to_json()) of.write(self._model.to_json())
# plot model # 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")) of.write(model_to_dot(self._model).create("dot", format="svg"))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment