From e19910eda5bf27c4c8d1139731af9b2d6ff23a2f Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Wed, 9 May 2018 11:24:09 +0200
Subject: [PATCH] Use yaml.safe_load for options to avoid getting unicode
strings
---
toolkit.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/toolkit.py b/toolkit.py
index 3d45c6e..7636dc6 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -7,6 +7,7 @@ if version_info[0] > 2:
import os
import json
+import yaml
import pickle
import importlib
import csv
@@ -133,7 +134,7 @@ class ClassificationProject(object):
def _init_from_dir(self, dirname):
with open(os.path.join(dirname, "options.json")) as f:
- options = json.load(f)
+ options = yaml.safe_load(f)
options["kwargs"]["project_dir"] = dirname
self._init_from_args(os.path.basename(dirname), *options["args"], **options["kwargs"])
--
GitLab