Add KubernetesPodOperator & DockerOperator management

cosmos/providers/dbt/__init__.py

@@ -2,28 +2,53 @@
 dbt support for Airflow. Contains dags, task groups, and operators.
 """
 
+from cosmos.providers.dbt.core.operators.docker import (
+    DbtLSDockerOperator,
+    DbtRunDockerOperator,
+    DbtRunOperationDockerOperator,
+    DbtSeedDockerOperator,
+    DbtTestDockerOperator,
+)
+from cosmos.providers.dbt.core.operators.kubernetes import (
+    DbtLSKubernetesOperator,
+    DbtRunKubernetesOperator,
+    DbtRunOperationKubernetesOperator,
+    DbtSeedKubernetesOperator,
+    DbtTestKubernetesOperator,
+)
+
 # re-export the operators
-from .core.operators import (
-    DbtLSOperator,
-    DbtRunOperationOperator,
-    DbtRunOperator,
-    DbtSeedOperator,
-    DbtTestOperator,
+from cosmos.providers.dbt.core.operators.local import (
+    DbtLSLocalOperator,
+    DbtRunLocalOperator,
+    DbtRunOperationLocalOperator,
+    DbtSeedLocalOperator,
+    DbtTestLocalOperator,
 )
 
 # re-export user facing utilities
-from .core.utils.data_aware_scheduling import get_dbt_dataset
+from cosmos.providers.dbt.core.utils.data_aware_scheduling import get_dbt_dataset
 
 # re-export the dag and task group
-from .dag import DbtDag
-from .task_group import DbtTaskGroup
+from cosmos.providers.dbt.dag import DbtDag
+from cosmos.providers.dbt.task_group import DbtTaskGroup
 
 __all__ = [
-    DbtLSOperator,
-    DbtRunOperationOperator,
-    DbtRunOperator,
-    DbtSeedOperator,
-    DbtTestOperator,
+    DbtLSLocalOperator,
+    DbtRunOperationLocalOperator,
+    DbtRunLocalOperator,
+    DbtSeedLocalOperator,
+    DbtTestLocalOperator,
+    DbtLSDockerOperator,
+    DbtRunOperationDockerOperator,
+    DbtRunDockerOperator,
+    DbtSeedDockerOperator,
+    DbtTestDockerOperator,
+    DbtLSKubernetesOperator,
+    DbtRunOperationKubernetesOperator,
+    DbtRunKubernetesOperator,
+    DbtSeedKubernetesOperator,
+    DbtTestKubernetesOperator,
     get_dbt_dataset,
     DbtDag,
     DbtTaskGroup,