fix: Use load_extension directly

m1so · m1so · commit 49419a7a655d · 2025-11-19T16:59:32.000+01:00
diff --git a/deepnote_toolkit/sql/duckdb_sql.py b/deepnote_toolkit/sql/duckdb_sql.py
@@ -58,7 +58,7 @@ def _get_duckdb_connection():
             import_extension(
                 name=extension_name, force_install=True, con=_DEEPNOTE_DUCKDB_CONNECTION
             )
-            _DEEPNOTE_DUCKDB_CONNECTION.execute(f"LOAD '{extension_name}'")
+            _DEEPNOTE_DUCKDB_CONNECTION.load_extension(extension_name)
 
         _set_sample_size(_DEEPNOTE_DUCKDB_CONNECTION, _DEFAULT_DUCKDB_SAMPLE_SIZE)
 

Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,7 @@ def _get_duckdb_connection():`
`58`	`58`	`import_extension(`
`59`	`59`	`name=extension_name, force_install=True, con=_DEEPNOTE_DUCKDB_CONNECTION`
`60`	`60`	`)`
`61`		`- _DEEPNOTE_DUCKDB_CONNECTION.execute(f"LOAD '{extension_name}'")`
	`61`	`+ _DEEPNOTE_DUCKDB_CONNECTION.load_extension(extension_name)`
`62`	`62`
`63`	`63`	`_set_sample_size(_DEEPNOTE_DUCKDB_CONNECTION, _DEFAULT_DUCKDB_SAMPLE_SIZE)`
`64`	`64`