Revert "Fix memory benchmarks"

This reverts commit 99d5bb2.
LSSTDESC · Oct 10, 2024 · 04203d6 · 04203d6
1 parent 99d5bb2
commit 04203d6
Showing 1 changed file with 10 additions and 12 deletions.
diff --git a/benchmarks/benchmarks.py b/benchmarks/benchmarks.py
@@ -65,18 +65,16 @@ def peakmem_learn_loop(ml_model):
         metrics_file = str(Path(dir_name) / "metrics.csv")
         output_queried_file = str(Path(dir_name) / "queried.csv")
         learn_loop(
-            LoopConfiguration(
-                nloops=25,
-                features_method="malanchev",
-                classifier=ml_model,
-                strategy="RandomSampling",
-                path_to_features=features_file,
-                output_metrics_file=metrics_file,
-                output_queried_file=output_queried_file,
-                training="original",
-                batch=1,
-            )
-        )
+            nloops=25,
+            features_method="malanchev",
+            classifier=ml_model,
+            strategy="RandomSampling",
+            path_to_features=features_file,
+            output_metrics_file=metrics_file,
+            output_queried_file=output_queried_file,
+            training="original",
+            batch=1,
+        )  
 
 
 # Parameterize the ML models and strategies we benchmark.