added allow_pickle=True to np.load()

autonomio · Aug 2, 2019 · 3fed99e · 3fed99e
1 parent 6a4fbfa
commit 3fed99e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/talos/commands/restore.py b/talos/commands/restore.py
@@ -33,7 +33,7 @@ def __init__(self, path_to_zip):
         z.extractall(self.extract_to)
 
         # add params dictionary
-        self.params = load(self.file_prefix + '_params.npy').item()
+        self.params = load(self.file_prefix + '_params.npy', allow_pickle=True).item()
 
         # add experiment details
         self.details = read_csv(self.file_prefix + '_details.txt', header=None)