catch error if dataframe does not exist

SimonBlanke · Dec 29, 2023 · 55ed026 · 55ed026
1 parent cb9dbd2
commit 55ed026
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/surfaces/machine_learning/machine_learning_functions/base_machine_learning_function.py b/surfaces/machine_learning/machine_learning_functions/base_machine_learning_function.py
@@ -50,7 +50,11 @@ def collect_data(self, if_exists="append"):
         self.sql_data.save(self.__name__, search_data, if_exists)
 
     def load_search_data(self):
-        return self.sql_data.load(self.__name__)
+        try:
+            dataframe = self.sql_data.load(self.__name__)
+        except:
+            print("Path 2 database: ", self.sql_data.path)
+        return dataframe
 
     def objective_function_dict(self, params):
         try: