Fix trajectory file output

dengdifan · Mar 8, 2021 · c95fbf3 · c95fbf3
1 parent 5d63668
commit c95fbf3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/autoPyTorch/api/base_task.py b/autoPyTorch/api/base_task.py
@@ -858,11 +858,11 @@ def _search(
                 saveable_trajectory = \
                     [list(entry[:2]) + [entry[2].get_dictionary()] + list(entry[3:])
                      for entry in self.trajectory]
-                if os.path.exists(trajectory_filename):
+                try:
                     with open(trajectory_filename, 'w') as fh:
                         json.dump(saveable_trajectory, fh)
-                else:
-                    self._logger.warning(f"File {trajectory_filename} does not exists...")
+                except Exception as e:
+                    self._logger.warning(f"Cannot save {trajectory_filename} due to {e}...")
             except Exception as e:
                 self._logger.exception(str(e))
                 raise