fix(pandas): force backend options registration on trace.enable() calls

ibis-project · Jul 6, 2022 · 8818fe6 · 8818fe6
1 parent 7b94a49
commit 8818fe6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/ibis/backends/dask/trace.py b/ibis/backends/dask/trace.py
@@ -3,6 +3,7 @@
 import traceback
 from datetime import datetime
 
+import ibis
 from ibis.backends.pandas.dispatcher import TwoLevelDispatcher
 from ibis.config import options
 from ibis.expr import types as ir
@@ -69,6 +70,10 @@ def add_one(v):
 
 def enable():
     """Enable tracing."""
+    if options.dask is None:
+        # dask options haven't been registered yet - force module __getattr__
+        ibis.dask
+
     options.dask.enable_trace = True
     logging.getLogger('ibis.dask.trace').setLevel(logging.DEBUG)
 

diff --git a/ibis/backends/pandas/trace.py b/ibis/backends/pandas/trace.py
@@ -3,6 +3,7 @@
 import traceback
 from datetime import datetime
 
+import ibis
 from ibis.backends.pandas.dispatcher import TwoLevelDispatcher
 from ibis.config import options
 from ibis.expr import types as ir
@@ -82,6 +83,9 @@ def add_one(v):
 
 def enable():
     """Enable tracing."""
+    if options.pandas is None:
+        # pandas options haven't been registered yet - force module __getattr__
+        ibis.pandas
     options.pandas.enable_trace = True
     logging.getLogger('ibis.backends.pandas.trace').setLevel(logging.DEBUG)