test: enable gemini-1.5-pro-002 tests (#1707)

Author: shuoweil

tests/system/load/test_llm.py

@@ -41,7 +41,7 @@ def llm_remote_text_df(session, llm_remote_text_pandas_df):
 @pytest.mark.parametrize(
     "model_name",
     (
-        # "gemini-1.5-pro-002",
+        "gemini-1.5-pro-002",
         "gemini-1.5-flash-002",
     ),
 )

tests/system/small/ml/test_llm.py

@@ -111,7 +111,7 @@ def test_create_load_multimodal_embedding_generator_model(
         "gemini-1.5-pro-preview-0514",
         "gemini-1.5-flash-preview-0514",
         "gemini-1.5-pro-001",
-        # "gemini-1.5-pro-002",
+        "gemini-1.5-pro-002",
         "gemini-1.5-flash-001",
         "gemini-1.5-flash-002",
         "gemini-2.0-flash-exp",
@@ -146,7 +146,7 @@ def test_create_load_gemini_text_generator_model(
         "gemini-1.5-pro-preview-0514",
         "gemini-1.5-flash-preview-0514",
         "gemini-1.5-pro-001",
-        # "gemini-1.5-pro-002",
+        "gemini-1.5-pro-002",
         "gemini-1.5-flash-001",
         "gemini-1.5-flash-002",
         "gemini-2.0-flash-exp",
@@ -173,7 +173,7 @@ def test_gemini_text_generator_predict_default_params_success(
         "gemini-1.5-pro-preview-0514",
         "gemini-1.5-flash-preview-0514",
         "gemini-1.5-pro-001",
-        # "gemini-1.5-pro-002",
+        "gemini-1.5-pro-002",
         "gemini-1.5-flash-001",
         "gemini-1.5-flash-002",
         "gemini-2.0-flash-exp",
@@ -202,7 +202,7 @@ def test_gemini_text_generator_predict_with_params_success(
         "gemini-1.5-pro-preview-0514",
         "gemini-1.5-flash-preview-0514",
         "gemini-1.5-pro-001",
-        # "gemini-1.5-pro-002",
+        "gemini-1.5-pro-002",
         "gemini-1.5-flash-001",
         "gemini-1.5-flash-002",
         "gemini-2.0-flash-exp",
@@ -233,7 +233,7 @@ def test_gemini_text_generator_multi_cols_predict_success(
         "gemini-1.5-pro-preview-0514",
         "gemini-1.5-flash-preview-0514",
         "gemini-1.5-pro-001",
-        # "gemini-1.5-pro-002",
+        "gemini-1.5-pro-002",
         "gemini-1.5-flash-001",
         "gemini-1.5-flash-002",
         "gemini-2.0-flash-exp",
@@ -756,7 +756,7 @@ def test_text_embedding_generator_retry_no_progress(session, bq_connection):
 @pytest.mark.parametrize(
     "model_name",
     (
-        # "gemini-1.5-pro-002",
+        "gemini-1.5-pro-002",
         "gemini-1.5-flash-002",
         "gemini-2.0-flash-001",
         "gemini-2.0-flash-lite-001",
@@ -786,7 +786,7 @@ def test_llm_gemini_score(llm_fine_tune_df_default_index, model_name):
 @pytest.mark.parametrize(
     "model_name",
     (
-        # "gemini-1.5-pro-002",
+        "gemini-1.5-pro-002",
         "gemini-1.5-flash-002",
         "gemini-2.0-flash-001",
         "gemini-2.0-flash-lite-001",

tests/system/small/ml/test_multimodal_llm.py

@@ -42,7 +42,7 @@ def test_multimodal_embedding_generator_predict_default_params_success(
     "model_name",
     (
         "gemini-1.5-pro-001",
-        # "gemini-1.5-pro-002",
+        "gemini-1.5-pro-002",
         "gemini-1.5-flash-001",
         "gemini-1.5-flash-002",
         "gemini-2.0-flash-exp",
@@ -71,7 +71,7 @@ def test_gemini_text_generator_multimodal_input(
     "model_name",
     (
         "gemini-1.5-pro-001",
-        # "gemini-1.5-pro-002",
+        "gemini-1.5-pro-002",
         "gemini-1.5-flash-001",
         "gemini-1.5-flash-002",
         "gemini-2.0-flash-exp",