Skip to content

Conversation

@google-labs-jules
Copy link
Contributor

Moved long-running ML tests to the large directory to improve test suite organization and execution time management. This involves creating new test files in tests/system/large/ml/ and appending to existing ones, while removing the corresponding tests from tests/system/small/ml/.


PR created automatically by Jules for task 11110816135219701367 started by @tswast

Moved the following long-running tests from `tests/system/small/ml/` to `tests/system/large/ml/`:
- `test_llm_gemini_score`
- `test_llm_gemini_pro_score_params`
- `test_gemini_text_generator_predict_default_params_success`
- `test_gemini_text_generator_predict_output_schema_success`
- `test_gemini_text_generator_multi_cols_predict_success`
- `test_gemini_text_generator_predict_with_params_success`
- `test_create_load_gemini_text_generator_model`
- `test_gemini_text_generator_multimodal_input`
- `test_linear_reg_model_global_explain`
@google-labs-jules
Copy link
Contributor Author

👋 Jules, reporting for duty! I'm here to lend a hand with this pull request.

When you start a review, I'll add a 👀 emoji to each comment to let you know I've read it. I'll focus on feedback directed at me and will do my best to stay out of conversations between you and other bots or reviewers to keep the noise down.

I'll push a commit with your requested changes shortly after. Please note there might be a delay between these steps, but rest assured I'm on the job!

For more direct control, you can switch me to Reactive Mode. When this mode is on, I will only act on comments where you specifically mention me with @jules. You can find this option in the Pull Request section of your global Jules UI settings. You can always switch back!


For security, I will only act on instructions from the user who triggered this task.

New to Jules? Learn more at jules.google/docs.

@product-auto-label product-auto-label bot added size: l Pull request size is large. api: bigquery Issues related to the googleapis/python-bigquery-dataframes API. labels Dec 10, 2025
Moved the following long-running tests from `tests/system/small/ml/` to `tests/system/large/ml/`:
- `test_llm_gemini_score`
- `test_llm_gemini_pro_score_params`
- `test_gemini_text_generator_predict_default_params_success`
- `test_gemini_text_generator_predict_output_schema_success`
- `test_gemini_text_generator_multi_cols_predict_success`
- `test_gemini_text_generator_predict_with_params_success`
- `test_create_load_gemini_text_generator_model`
- `test_gemini_text_generator_multimodal_input`
- `test_linear_reg_model_global_explain`
@tswast tswast changed the title Move long-running LLM tests to the large directory chore: Move long-running LLM tests to the large directory Dec 10, 2025
Moved the following long-running tests from `tests/system/small/ml/` to `tests/system/large/ml/`:
- `test_llm_gemini_score`
- `test_llm_gemini_pro_score_params`
- `test_gemini_text_generator_predict_default_params_success`
- `test_gemini_text_generator_predict_output_schema_success`
- `test_gemini_text_generator_multi_cols_predict_success`
- `test_gemini_text_generator_predict_with_params_success`
- `test_create_load_gemini_text_generator_model`
- `test_gemini_text_generator_multimodal_input`
- `test_linear_reg_model_global_explain`
Copy link
Collaborator

Choose a reason for hiding this comment

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

tests/system/small/ml/test_llm.py:19:1: F401 'pyarrow as pa' imported but unused

Please remove this unused import.

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Fixed. I've removed the unused pyarrow as pa import.

Moved the following long-running tests from `tests/system/small/ml/` to `tests/system/large/ml/`:
- `test_llm_gemini_score`
- `test_llm_gemini_pro_score_params`
- `test_gemini_text_generator_predict_default_params_success`
- `test_gemini_text_generator_predict_output_schema_success`
- `test_gemini_text_generator_multi_cols_predict_success`
- `test_gemini_text_generator_predict_with_params_success`
- `test_create_load_gemini_text_generator_model`
- `test_gemini_text_generator_multimodal_input`
- `test_linear_reg_model_global_explain`
@tswast tswast marked this pull request as ready for review December 10, 2025 22:37
@tswast tswast requested review from a team as code owners December 10, 2025 22:37
@tswast tswast requested a review from shuoweil December 10, 2025 22:37
@shuoweil shuoweil enabled auto-merge (squash) December 10, 2025 22:49
@tswast tswast added the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Dec 10, 2025
@bigframes-bot bigframes-bot removed the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Dec 10, 2025
@shuoweil shuoweil merged commit 29bf680 into main Dec 10, 2025
19 of 26 checks passed
@shuoweil shuoweil deleted the chore-move-long-running-llm-tests-11110816135219701367 branch December 10, 2025 23:30
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

api: bigquery Issues related to the googleapis/python-bigquery-dataframes API. size: l Pull request size is large.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants