skip srsr/depression only for Buhos

CoLRev-Environment · Oct 14, 2024 · e94c118 · e94c118
1 parent da801d1
commit e94c118
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/notebooks/evaluation.py b/notebooks/evaluation.py
@@ -311,9 +311,6 @@ def append_to_output(result: dict, *, package_name: str) -> None:
     for benchmark_path in evaluation.get_dataset_labels():
         print(f"Dataset: {benchmark_path}")
 
-        if benchmark_path in ["srsr", "depression"]:
-            continue
-
         dedupe_benchmark = DedupeBenchmarker(
             benchmark_path=Path(f"data/{benchmark_path}")
         )
@@ -367,6 +364,8 @@ def append_to_output(result: dict, *, package_name: str) -> None:
 
         print()
 
+        if benchmark_path in ["srsr", "depression"]:
+            continue
         # Buhos
         records_df.to_csv("notebooks/buhos/records.csv", index=False)
         timestamp = datetime.now()