Replacing hardcoded port for a connection attr reference

astronomer · Oct 2, 2023 · bcb65a8 · bcb65a8
1 parent fabedd8
commit bcb65a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cosmos/profiles/postgres/user_pass.py b/cosmos/profiles/postgres/user_pass.py
@@ -40,7 +40,7 @@ class PostgresUserPasswordProfileMapping(BaseProfileMapping):
     def profile(self) -> dict[str, Any | None]:
         "Gets profile. The password is stored in an environment variable."
         profile = {
-            "port": 5432,
+            "port": self.conn.port,
             **self.mapped_params,
             **self.profile_args,
             # password should always get set as env var