Fix report single metric (#585)

VectorInstitute · Mar 15, 2024 · 31a1480 · 31a1480
1 parent dff88de
commit 31a1480
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/cyclops/report/report.py b/cyclops/report/report.py
@@ -1096,9 +1096,15 @@ def export(
         else:
             today_now = dt_datetime.now().strftime("%Y-%m-%d %H:%M:%S")
 
-        current_report_metrics: List[List[PerformanceMetric]] = []
+        current_report_metrics: Union[
+            List[List[PerformanceMetric]], List[PerformanceMetric]
+        ] = []
         sweep_metrics(self._model_card, current_report_metrics)
-        current_report_metrics_set = current_report_metrics[0]
+        current_report_metrics_set = (
+            current_report_metrics[0]
+            if isinstance(current_report_metrics[0], list)
+            else [current_report_metrics[0]]
+        )
 
         report_paths = glob.glob(
             os.path.join(