Fix imports in stage

LSSTDESC · Jul 29, 2024 · f781b99 · f781b99
1 parent d56f4b3
commit f781b99
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/rail/core/stage.py b/src/rail/core/stage.py
@@ -5,7 +5,8 @@
 import yaml
 from math import ceil
 
-from ceci import PipelineStage, MiniPipeline
+from ceci.stage import PipelineStage
+from ceci.pipeline import MiniPipeline
 from ceci.config import StageParameter as Param
 from rail.core.data import DATA_STORE, DataHandle