add ability to specify host/port for Snowflake connection

astronomer · Jun 24, 2024 · 1c67479 · 1c67479
1 parent 5f5616c
commit 1c67479
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cosmos/profiles/snowflake/user_pass.py b/cosmos/profiles/snowflake/user_pass.py
@@ -40,6 +40,8 @@ class SnowflakeUserPasswordProfileMapping(BaseProfileMapping):
         "warehouse": "extra.warehouse",
         "schema": "schema",
         "role": "extra.role",
+        "host": "extra.host",
+        "port": "extra.port",
     }
 
     def can_claim_connection(self) -> bool: