Skip to content

Commit 3867390

Browse files
authored
test: explicitly use US location for session in tests (#650)
This avoids some warnings we see and ignore in our tests. It might also address some flakiness in `tests/system/small/ml/test_llm.py::test_create_text_generator_model` and `tests/system/small/ml/test_llm.py::test_create_text_generator_32k_model`, but the root cause of that flakiness is still TBD.
1 parent c67e501 commit 3867390

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/system/conftest.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,10 @@ def resourcemanager_client(
128128

129129
@pytest.fixture(scope="session")
130130
def session() -> bigframes.Session:
131-
return bigframes.Session()
131+
context = bigframes.BigQueryOptions(
132+
location="US",
133+
)
134+
return bigframes.Session(context=context)
132135

133136

134137
@pytest.fixture(scope="session")

0 commit comments

Comments
 (0)