Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Address review comments on PR 1347 regarding profile_config for ExecutionMode.Docker #1413

Merged
merged 7 commits into from
Dec 20, 2024
Next Next commit
adds profile_config for Docker
andrewhlui committed Nov 27, 2024

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
commit 5cfdce3becd173e925519c0c47ab8c14aee4c4f9
3 changes: 3 additions & 0 deletions cosmos/operators/docker.py
Original file line number Diff line number Diff line change
@@ -4,6 +4,7 @@

from airflow.utils.context import Context

from cosmos.config import ProfileConfig
from cosmos.operators.base import (
AbstractDbtBaseOperator,
DbtBuildMixin,
@@ -42,8 +43,10 @@ class DbtDockerBaseOperator(AbstractDbtBaseOperator, DockerOperator): # type: i
def __init__(
self,
image: str, # Make image a required argument since it's required by DockerOperator
profile_config: ProfileConfig | None = None,
**kwargs: Any,
) -> None:
self.profile_config = profile_config
super().__init__(image=image, **kwargs)

def build_and_run_cmd(self, context: Context, cmd_flags: list[str] | None = None) -> Any: