Merge pull request #58 from OscarAdserballe/support_connections

Add support for connections to remote models
kristeligt-dagblad · Aug 6, 2024 · 5ef1535 · 5ef1535
2 parents f7001ad + 40721ff
commit 5ef1535
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/dbt_project.yml b/dbt_project.yml
@@ -1,5 +1,5 @@
 name: "dbt_ml"
-version: "0.6.0"
+version: "0.6.1"
 
 config-version: 2
 

diff --git a/macros/materializations/model.sql b/macros/materializations/model.sql
@@ -58,14 +58,20 @@
     {{ sql_header if sql_header is not none }}
 
     create or replace model {{ relation }}
+
+    {% if ml_config.get('connection_name') %}
+        remote with connection `{{ ml_config.pop('connection_name') }}`
+    {% endif %}
+
     {{ dbt_ml.model_options(
         ml_config=ml_config,
         labels=raw_labels
     ) }}
-    {%- if ml_config.get('MODEL_TYPE', ml_config.get('model_type', '')).lower() != 'tensorflow' -%}
-    as (
-        {{ sql }}
-    );
+
+    {%- if sql -%}
+        as (
+            {{ sql }}
+        );
     {%- endif -%}
 {% endmacro %}