add load_predictions option

autogluon · Feb 5, 2025 · b03976d · b03976d
1 parent 7e7844c
commit b03976d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tabrepo/repository/evaluation_repository.py b/tabrepo/repository/evaluation_repository.py
@@ -186,8 +186,8 @@ def _construct_config_scorer(self,
             raise ValueError(f'Invalid config_scorer_type: {config_scorer_type}')
 
     @classmethod
-    def from_context(cls, version: str = None, cache: bool = False, prediction_format: str = "memmap") -> Self:
-        return load_repository(version=version, cache=cache, prediction_format=prediction_format)
+    def from_context(cls, version: str = None, cache: bool = False, load_predictions: bool = True, prediction_format: str = "memmap") -> Self:
+        return load_repository(version=version, cache=cache, load_predictions=load_predictions, prediction_format=prediction_format)
 
     # TODO: 1. Cleanup results_lst_simulation_artifacts, 2. Make context work with tasks instead of datasets x folds
     # TODO: Get raw data from repo method (X, y)