update how we import

allenai · Nov 20, 2024 · 5b1ab9a · 5b1ab9a
1 parent 4554054
commit 5b1ab9a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/rslp/forest_loss_driver/inference/__init__.py b/rslp/forest_loss_driver/inference/__init__.py
@@ -1,6 +1,7 @@
 """Inference pipeline steps for the forest loss driver."""
 
 from .best_image_selector import select_best_images_pipeline
+from .config import PredictPipelineConfig
 from .extract_alerts import extract_alerts_pipeline
 from .materialize_dataset import materialize_forest_loss_driver_dataset
 from .model_predict import forest_loss_driver_model_predict
@@ -10,4 +11,5 @@
     "extract_alerts_pipeline",
     "materialize_forest_loss_driver_dataset",
     "forest_loss_driver_model_predict",
+    "PredictPipelineConfig",
 ]
diff --git a/rslp/forest_loss_driver/predict_pipeline.py b/rslp/forest_loss_driver/predict_pipeline.py
@@ -5,11 +5,13 @@
 
 from rslp.log_utils import get_logger
 
-from .inference.best_image_selector import select_best_images_pipeline
-from .inference.config import PredictPipelineConfig
-from .inference.extract_alerts import extract_alerts_pipeline
-from .inference.materialize_dataset import materialize_forest_loss_driver_dataset
-from .inference.model_predict import forest_loss_driver_model_predict
+from .inference import (
+    PredictPipelineConfig,
+    extract_alerts_pipeline,
+    forest_loss_driver_model_predict,
+    materialize_forest_loss_driver_dataset,
+    select_best_images_pipeline,
+)
 
 logger = get_logger(__name__)