fix(snowflake): add generation oddrn for models

opendatadiscovery · Sep 21, 2023 · 1fd6558 · 1fd6558
1 parent a41a397
commit 1fd6558
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/odd_dbt/mapper/generator.py b/odd_dbt/mapper/generator.py
@@ -62,7 +62,7 @@ class SnowflakeGenerator(Generator):
     def __init__(self, credentials: Credentials) -> None:
         self.credentials = credentials
 
-    def _get_oddrn_for(self, model: ModelNode) -> str:
+    def _get_oddrn_for_model(self, model: ModelNode) -> str:
         host = f"{self.credentials['account'].upper()}.snowflakecomputing.com"
         database = self.credentials["database"] or self.credentials["dbname"]
         database = database.upper()

diff --git a/odd_dbt/mapper/lineage.py b/odd_dbt/mapper/lineage.py
@@ -99,7 +99,7 @@ def map_model(self, node: ModelNode) -> ModelEntity:
             type=DataEntityType.JOB,
             metadata=[get_model_metadata(node)],
         )
-        model_entity.add_output(get_materialized_entity_oddrn(node, self._context))
+        model_entity.add_input(get_materialized_entity_oddrn(node, self._context))
         return model_entity