Updated user_pass.py

astronomer · Nov 25, 2023 · 080c2dd · 080c2dd
1 parent 472f14e
commit 080c2dd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cosmos/profiles/postgres/user_pass.py b/cosmos/profiles/postgres/user_pass.py
@@ -46,6 +46,9 @@ def profile(self) -> dict[str, Any | None]:
             "password": self.get_env_var_format("password"),
         }
 
+        if "schema" in self.profile_args:
+            profile["schema"] = self.get_env_var_format("schema")
+
         return self.filter_null(profile)
 
     @property