diff --git a/src/databricks/labs/dqx/workflows_runner.py b/src/databricks/labs/dqx/workflows_runner.py index a1b27c69b..dca6499dd 100644 --- a/src/databricks/labs/dqx/workflows_runner.py +++ b/src/databricks/labs/dqx/workflows_runner.py @@ -59,11 +59,13 @@ def all(cls, config: WorkspaceConfig) -> "WorkflowsRunner": spark_conf=config.e2e_spark_conf, override_clusters=config.e2e_override_clusters, ) - anomaly_trainer = AnomalyTrainerWorkflow( - spark_conf=config.anomaly_spark_conf, - override_clusters=config.anomaly_override_clusters, - ) - workflows: list[Workflow] = [profiler, quality_checker, e2e, anomaly_trainer] + workflows: list[Workflow] = [profiler, quality_checker, e2e] + if ANOMALY_AVAILABLE: + anomaly_trainer = AnomalyTrainerWorkflow( + spark_conf=config.anomaly_spark_conf, + override_clusters=config.anomaly_override_clusters, + ) + workflows.append(anomaly_trainer) return cls(workflows)