fix ensemble edge case

views-platform · Dec 12, 2024 · f4ed737 · f4ed737
1 parent 70098e6
commit f4ed737
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/views_pipeline_core/managers/ensemble.py b/views_pipeline_core/managers/ensemble.py
@@ -13,6 +13,8 @@
 import subprocess
 from datetime import datetime
 import pandas as pd
+from views_pipeline_core.files.utils import save_dataframe, read_dataframe
+from views_pipeline_core.configs.pipeline import PipelineConfig
 
 logger = logging.getLogger(__name__)
 

diff --git a/views_pipeline_core/managers/model.py b/views_pipeline_core/managers/model.py
@@ -581,8 +581,9 @@ def __init__(self, model_path: ModelPathManager) -> None:
                 "config_sweep.py", "get_sweep_config"
             )
         self.set_dataframe_format(format=".parquet")
-        from views_pipeline_core.data.dataloaders import ViewsDataLoader
-        self._data_loader = ViewsDataLoader(model_path=self._model_path)
+        if self._model_path.target == "model":
+            from views_pipeline_core.data.dataloaders import ViewsDataLoader
+            self._data_loader = ViewsDataLoader(model_path=self._model_path)
 
 
     def set_dataframe_format(self, format: str) -> None: