We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 48f9953 commit cc20328Copy full SHA for cc20328
1 file changed
packages/bigframes/tests/unit/core/compile/sqlglot/tpch/conftest.py
@@ -157,7 +157,9 @@ def read_gbq_table_no_snapshot(*args, **kwargs):
157
kwargs["enable_snapshot"] = False
158
return original_read_gbq_table(*args, **kwargs)
159
160
- session._loader.read_gbq_table = read_gbq_table_no_snapshot
161
-
162
session._executor = compiler_session.SQLCompilerExecutor()
163
- return session
+
+ with mock.patch.object(
+ session._loader, "read_gbq_table", new=read_gbq_table_no_snapshot
164
+ ):
165
+ yield session
0 commit comments