diff --git a/talos/commands/deploy.py b/talos/commands/deploy.py index eb4af0c4..73c71216 100644 --- 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 index 738b4063..daf05ab6 100644 --- 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',