pyarrow should not be required to run dbt-duckdb python models

duckdb · May 4, 2023 · 17cb912 · 17cb912
1 parent 4ddd363
commit 17cb912
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/dbt/include/duckdb/macros/adapters.sql b/dbt/include/duckdb/macros/adapters.sql
@@ -76,10 +76,11 @@
 def materialize(df, con):
     try:
         import pyarrow
+        pyarrow_available = True
     except ImportError:
-        pass
+        pyarrow_available = False
     finally:
-        if isinstance(df, pyarrow.Table):
+        if pyarrow_available and isinstance(df, pyarrow.Table):
             # https://github.com/duckdb/duckdb/issues/6584
             import pyarrow.dataset
     con.execute('create table {{ relation.include(database=adapter.use_database()) }} as select * from df')