Fix bug extend logging.

aai-institute · Apr 27, 2024 · 760d4da · 760d4da
1 parent daf8230
commit 760d4da
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/scripts/evaluate_curves.py b/scripts/evaluate_curves.py
@@ -89,7 +89,7 @@ def _evaluate_curves(
         / str(repetition_id)
         / valuation_method_name
     )
-    if os.path.exists(output_dir / f"{curve_name}.curve.csv"):
+    if os.path.exists(output_dir / f"{curve_name}.csv"):
         return logger.info(f"Curve data exists in '{output_dir}'. Skipping...")
 
     values = Accessor.valuation_results(

diff --git a/scripts/run_pipeline.py b/scripts/run_pipeline.py
@@ -141,11 +141,10 @@ def run_pipeline():
                         "curves"
                     ].keys():
                         logger.info(
-                            f"Calculate metric {curve_name} for dataset {dataset_name}, "
+                            f"Calculate curve {curve_name} for dataset {dataset_name}, "
                             f"valuation method {valuation_method_name} and seed "
-                            f"{repetition_id}."
+                            f"{repetition_id} on model {model_name}."
                         )
-                        logger.info(f"Evaluate metric {curve_name}.")
                         repeat(
                             _evaluate_curves,
                             experiment_name,
@@ -163,7 +162,7 @@ def run_pipeline():
                         logger.info(
                             f"Calculate metric {metric_name} for dataset {dataset_name}, "
                             f"valuation method {valuation_method_name} and seed "
-                            f"{repetition_id}."
+                            f"{repetition_id} on model {model_name}."
                         )
                         logger.info(f"Evaluate metric {metric_name}.")
                         repeat(