Skip to content
This repository was archived by the owner on May 7, 2026. It is now read-only.

test: explicitly use US location for session in tests#650

Merged
tswast merged 1 commit intomainfrom
tswast-test_create_text_generator_model
May 1, 2024
Merged

test: explicitly use US location for session in tests#650
tswast merged 1 commit intomainfrom
tswast-test_create_text_generator_model

Conversation

@tswast
Copy link
Copy Markdown
Contributor

@tswast tswast commented May 1, 2024

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.

Thank you for opening a Pull Request! Before submitting your PR, there are a few things you can do to make sure it goes smoothly:

  • Make sure to open an issue as a bug/issue before writing your code! That way we can discuss the change, evaluate designs, and agree on the general idea
  • Ensure the tests and linter pass
  • Code coverage does not decrease (if any source code was changed)
  • Appropriate docs were updated (if necessary)

Fixes #<issue_number_goes_here> 🦕

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.
@tswast tswast requested review from a team and ashleyxuu May 1, 2024 18:29
@product-auto-label product-auto-label Bot added the size: xs Pull request size is extra small. label May 1, 2024
@product-auto-label product-auto-label Bot added the api: bigquery Issues related to the googleapis/python-bigquery-dataframes API. label May 1, 2024
Comment thread tests/system/conftest.py
def session() -> bigframes.Session:
return bigframes.Session()
context = bigframes.BigQueryOptions(
location="US",
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This should already be the default

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Note that the line below that raises a warning. So this avoids that noise at least, even if it doesn't fix the issue.

@tswast tswast merged commit 3867390 into main May 1, 2024
@tswast tswast deleted the tswast-test_create_text_generator_model branch May 1, 2024 21:26
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

api: bigquery Issues related to the googleapis/python-bigquery-dataframes API. size: xs Pull request size is extra small.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants