<filename>src/icolos/utils/enums/flow_control_enums.py
from icolos.core.workflow_steps.prediction.active_learning import StepActiveLearning
from icolos.utils.enums.step_enums import StepBaseEnum
from icolos.core.flow_control.iterator import StepIterator

_SBE = StepBaseEnum


class FlowControlInitializationEnum:
    # These steps are responsible for initializing other steps as part of their execution
    # Keep these separate to the main pool of steps to avoid circular imports

    FLOW_CONTROL_INIT_DICT = {
        _SBE.STEP_ITERATOR: StepIterator,
        _SBE.STEP_ACTIVE_LEARNING: StepActiveLearning,
    }

<filename>src/icolos/utils/enums/flow_control_enums.py
from icolos.core.workflow_steps.prediction.active_learning import StepActiveLearning
from icolos.utils.enums.step_enums import StepBaseEnum
from icolos.core.flow_control.iterator import StepIterator

_SBE = StepBaseEnum


class FlowControlInitializationEnum:
    # These steps are responsible for initializing other steps as part of their execution
    # Keep these separate to the main pool of steps to avoid circular imports

    FLOW_CONTROL_INIT_DICT = {
        _SBE.STEP_ITERATOR: StepIterator,
        _SBE.STEP_ACTIVE_LEARNING: StepActiveLearning,
    }

# These steps are responsible for initializing other steps as part of their execution # Keep these separate to the main pool of steps to avoid circular imports