Skip to content

Commit eec5328

Browse files
jcristcpcloud
authored andcommitted
fix(datafusion): fix creation of SessionContext in datafusion 40.1.0
1 parent 4aa402a commit eec5328

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

ibis/backends/datafusion/__init__.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,11 @@
4040
except ImportError:
4141
SessionConfig = None
4242

43+
try:
44+
from datafusion import RuntimeConfig
45+
except ImportError:
46+
RuntimeConfig = None
47+
4348
if TYPE_CHECKING:
4449
import pandas as pd
4550
import polars as pl
@@ -105,7 +110,12 @@ def do_connect(
105110
).with_information_schema(True)
106111
else:
107112
df_config = None
108-
self.con = SessionContext(df_config)
113+
if RuntimeConfig is None:
114+
self.con = SessionContext(df_config)
115+
else:
116+
# datafusion 40.1.0 has a bug where SessionContext requires
117+
# both SessionConfig and RuntimeConfig be provided.
118+
self.con = SessionContext(df_config, RuntimeConfig())
109119

110120
self._register_builtin_udfs()
111121

0 commit comments

Comments
 (0)