fixed an issue in Retore

Should fix #351.
autonomio · Jul 16, 2019 · 42951e9 · 42951e9
1 parent 21c52fa
commit 42951e9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/talos/commands/deploy.py b/talos/commands/deploy.py
@@ -69,10 +69,12 @@ def save_data(self):
 
         import pandas as pd
 
+        # input data is <= 2d
         try:
             x = pd.DataFrame(self.scan_object.x[:100])
             y = pd.DataFrame(self.scan_object.y[:100])
 
+        # input data is > 2d
         except ValueError:
             x = pd.DataFrame()
             y = pd.DataFrame()

diff --git a/talos/commands/restore.py b/talos/commands/restore.py
@@ -30,7 +30,8 @@ def __init__(self, path_to_zip):
         z.extractall(self.extract_to)
 
         # add params dictionary
-        self.params = np.load(self.file_prefix + '_params.npy').item()
+        self.params = np.load(self.file_prefix + '_params.npy',
+                              allow_pickle=True).item()
 
         # add experiment details
         self.details = pd.read_csv(self.file_prefix + '_details.txt',