remove unused import

SimonBlanke · May 26, 2024 · 24025d5 · 24025d5
1 parent 665dd98
commit 24025d5
Showing 1 changed file with 3 additions and 17 deletions.
diff --git a/collect_search_data.py b/collect_search_data.py
@@ -1,26 +1,12 @@
-from surfaces.test_functions import mathematical_functions, machine_learning_functions
+from surfaces.test_functions import machine_learning_functions
 from surfaces.data_collector import SurfacesDataCollector
 
 
-for mathematical_function in mathematical_functions:
-    print("\n ---", mathematical_function.name, "---")
-    try:
-        test_function = mathematical_function(metric="loss")
-    except TypeError:
-        test_function = mathematical_function(n_dim=2, metric="loss")
-
-    objective_function = test_function.objective_function
-    search_space = test_function.search_space(value_types="array")
-
-    sdc = SurfacesDataCollector()
-    sdc.collect(objective_function, search_space)
-
-
 for machine_learning_function in machine_learning_functions:
-    test_function = machine_learning_function(load_search_data=False)
+    test_function = machine_learning_function(evaluate_from_data=False)
 
     objective_function = test_function.objective_function
-    search_space = test_function.search_space(value_types="array")
+    search_space = test_function.search_space()
 
     sdc = SurfacesDataCollector()
     sdc.collect(objective_function, search_space)