run linter

Signed-off-by: Marta Stepniewska-Dziubinska <martas@nvidia.com>

Author: marta-sd

packages/nemo-evaluator-launcher/src/nemo_evaluator_launcher/exporters/base.py

@@ -352,7 +352,11 @@ def prepare_data_for_export(self, job_data: JobData) -> DataForExport | None:
             return None
 
         try:
-            metrics = extract_accuracy_metrics(artifacts_dir, metric_sep=self.config.metric_sep, include_task_name=self.config.include_task_name)
+            metrics = extract_accuracy_metrics(
+                artifacts_dir,
+                metric_sep=self.config.metric_sep,
+                include_task_name=self.config.include_task_name,
+            )
             harness, task = load_benchmark_info(artifacts_dir)
             container = job_data.data.get("eval_image", None)
             model_id = get_model_id(artifacts_dir)

packages/nemo-evaluator-launcher/src/nemo_evaluator_launcher/exporters/utils.py

@@ -143,7 +143,10 @@ class MetricConflictError(Exception):
 
 
 def extract_accuracy_metrics(
-    artifacts_dir: Path, log_metrics: List[str] = None, metric_sep: str = "_", include_task_name: bool = True
+    artifacts_dir: Path,
+    log_metrics: List[str] = None,
+    metric_sep: str = "_",
+    include_task_name: bool = True,
 ) -> Dict[str, float]:
     """Extract accuracy metrics from job results.
     artifacts_dir: Path to the artifacts directory
@@ -155,7 +158,11 @@ def extract_accuracy_metrics(
     if not artifacts_dir or not artifacts_dir.exists():
         raise RuntimeError(f"Artifacts directory {artifacts_dir} not found")
 
-    metrics = _extract_from_results_yml(artifacts_dir / RESULTS_FILE, metric_sep=metric_sep, include_task_name=include_task_name)
+    metrics = _extract_from_results_yml(
+        artifacts_dir / RESULTS_FILE,
+        metric_sep=metric_sep,
+        include_task_name=include_task_name,
+    )
     if not log_metrics:
         return metrics
 
@@ -559,14 +566,21 @@ def copy_artifacts(
 # =============================================================================
 
 
-def _extract_metrics_from_results(results: dict, metric_sep: str = "_", include_task_name: bool = True) -> Dict[str, float]:
+def _extract_metrics_from_results(
+    results: dict, metric_sep: str = "_", include_task_name: bool = True
+) -> Dict[str, float]:
     """Extract metrics from a 'results' dict (with optional 'groups'/'tasks')."""
     metrics: Dict[str, float] = {}
     for section in ["groups", "tasks"]:
         section_data = results.get(section)
         if isinstance(section_data, dict):
             for task_name, task_data in section_data.items():
-                task_metrics = _extract_task_metrics(task_name, task_data, metric_sep=metric_sep, include_task_name=include_task_name)
+                task_metrics = _extract_task_metrics(
+                    task_name,
+                    task_data,
+                    metric_sep=metric_sep,
+                    include_task_name=include_task_name,
+                )
                 _safe_update_metrics(
                     target=metrics,
                     source=task_metrics,
@@ -591,18 +605,28 @@ def _extract_from_results_yml(
     if "results" not in data:
         raise ValueError(f"Failed to parse {results_yml} - no results section found")
 
-    return _extract_metrics_from_results(data["results"], metric_sep=metric_sep, include_task_name=include_task_name)
+    return _extract_metrics_from_results(
+        data["results"], metric_sep=metric_sep, include_task_name=include_task_name
+    )
 
 
-def _extract_task_metrics(task_name: str, task_data: dict, metric_sep: str = "_", include_task_name: bool = True) -> Dict[str, float]:
+def _extract_task_metrics(
+    task_name: str,
+    task_data: dict,
+    metric_sep: str = "_",
+    include_task_name: bool = True,
+) -> Dict[str, float]:
     """Extract metrics from a task's metrics data."""
     extracted = {}
 
     metrics_data = task_data.get("metrics", {})
     if "groups" in task_data and task_data["groups"] is not None:
         for group_name, group_data in task_data["groups"].items():
             group_extracted = _extract_task_metrics(
-                f"{task_name}{metric_sep}{group_name}" if include_task_name else group_name, group_data
+                f"{task_name}{metric_sep}{group_name}"
+                if include_task_name
+                else group_name,
+                group_data,
             )
             _safe_update_metrics(
                 target=extracted,