minor update

autogluon · Dec 17, 2024 · ed1f2d1 · ed1f2d1
1 parent e824683
commit ed1f2d1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/autogluon/bench/eval/evaluation/benchmark_evaluator.py b/src/autogluon/bench/eval/evaluation/benchmark_evaluator.py
@@ -163,9 +163,10 @@ def _load_results(
         else:
             assert isinstance(paths, pd.DataFrame)
             results_raw = paths
-        with warnings.catch_warnings():
-            warnings.filterwarnings("ignore")
-            results_raw.loc[results_raw[TIME_INFER_S] == 0, TIME_INFER_S] = 0.001
+        if TIME_INFER_S in results_raw:
+            with warnings.catch_warnings():
+                warnings.filterwarnings("ignore")
+                results_raw.loc[results_raw[TIME_INFER_S] == 0, TIME_INFER_S] = 0.001
         if clean_data:
             # FIXME: This doesn't work on new tasks due to not comprehensive metadata
             results_raw = self._clean_data(results_raw=results_raw)