Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions DashAI/back/initial_components.py
Original file line number Diff line number Diff line change
Expand Up @@ -350,12 +350,16 @@
from DashAI.back.tasks.translation_task import TranslationTask

# Units
from DashAI.back.units.apply_converter_unit import ApplyConverterUnit
from DashAI.back.units.build_model_unit import BuildModelUnit
from DashAI.back.units.evaluate_model_unit import EvaluateModelUnit
from DashAI.back.units.fit_converter_unit import FitConverterUnit
from DashAI.back.units.fit_model_unit import FitModelUnit
from DashAI.back.units.load_dataset_unit import LoadDatasetUnit
from DashAI.back.units.prepare_and_split_unit import PrepareAndSplitUnit
from DashAI.back.units.save_dataset_unit import SaveDatasetUnit
from DashAI.back.units.save_model_unit import SaveModelUnit
from DashAI.back.units.transform_dataset_unit import TransformDatasetUnit

logging.basicConfig(level=logging.DEBUG)
log = logging.getLogger(__name__)
Expand Down Expand Up @@ -528,6 +532,10 @@ def get_initial_components():
FitModelUnit,
EvaluateModelUnit,
SaveModelUnit,
ApplyConverterUnit,
FitConverterUnit,
TransformDatasetUnit,
SaveDatasetUnit,
# Explainers
ContrastiveShap,
DiceCounterfactual,
Expand Down
Loading
Loading