Skip to content

Commit a441205

Browse files
chore: move long running LLM tests to the large directory
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`
1 parent 176adbb commit a441205

File tree

3 files changed

+1
-3
lines changed

3 files changed

+1
-3
lines changed

tests/system/large/ml/test_linear_model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -453,6 +453,7 @@ def test_model_centroids_with_custom_index(penguins_df_default_index):
453453
# If this line executes without errors, the model has correctly ignored the custom index columns
454454
model.predict(X_train.reset_index(drop=True))
455455

456+
456457
def test_linear_reg_model_global_explain(
457458
penguins_linear_model_w_global_explain, new_penguins_df
458459
):

tests/system/large/ml/test_llm.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
import pyarrow as pa
1717
import pytest
1818

19-
from bigframes import exceptions
2019
from bigframes.ml import llm
2120
import bigframes.pandas as bpd
2221
from bigframes.testing import utils

tests/system/large/ml/test_multimodal_llm.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,6 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
import pandas as pd
16-
import pyarrow as pa
1715
import pytest
1816

1917
from bigframes.ml import llm

0 commit comments

Comments
 (0)