From 5cfdce3becd173e925519c0c47ab8c14aee4c4f9 Mon Sep 17 00:00:00 2001
From: Andrew Lui <andrew.lui@gusto.com>
Date: Wed, 27 Nov 2024 17:50:39 -0500
Subject: [PATCH] adds profile_config for Docker

---
 cosmos/operators/docker.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/cosmos/operators/docker.py b/cosmos/operators/docker.py
index 05671b4d0..81b7196b3 100644
--- a/cosmos/operators/docker.py
+++ b/cosmos/operators/docker.py
@@ -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: