updating testing files

Author: CadillacBurgess1

generative_ai/model_garden/claude_3_batch_prediciton_bq.py

@@ -18,7 +18,7 @@
 output_uri = "bq://storage-samples.generative_ai.gen_ai_batch_prediction.predictions"
 
 
-def batch_predict_gemini_createjob(output_uri: str) -> str:
+def batch_predict_createjob(output_uri: str) -> str:
     """Perform batch text prediction using a Gemini AI model and returns the output location"""
 
     # [START generativeaionvertexai_batch_predict_anthropic_gemini_createjob_bigquery]
@@ -68,4 +68,4 @@ def batch_predict_gemini_createjob(output_uri: str) -> str:
 
 
 if __name__ == "__main__":
-    batch_predict_gemini_createjob(output_uri)
+    batch_predict_createjob(output_uri)

generative_ai/model_garden/claude_3_batch_prediction_gcs.py

@@ -17,7 +17,7 @@
 
 PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT")
 
-output_uri = "gs://python-docs-samples-tests"
+output_uri = "gs://kellysun-test-project-europe-west1"
 
 
 def batch_predict_createjob(output_uri: str) -> str:

generative_ai/model_garden/noxfile_config.py

@@ -22,7 +22,7 @@
 
 TEST_CONFIG_OVERRIDE = {
     # You can opt out from the test for specific Python versions.
-    "ignored_versions": ["2.7", "3.7", "3.8", "3.10", "3.11", "3.12"],
+    "ignored_versions": ["2.7", "3.7", "3.8", "3.10", "3.11"],
     # Old samples are opted out of enforcing Python type hints
     # All new samples should feature them
     "enforce_type_hints": True,

generative_ai/model_garden/requirements.txt

@@ -5,7 +5,7 @@ pillow==10.3.0; python_version < '3.8'
 pillow==10.3.0; python_version >= '3.8'
 google-cloud-aiplatform[all]==1.71.0
 sentencepiece==0.2.0
-google-auth==2.38.0
+google-auth==2.29.0
 anthropic[vertex]==0.28.0
 langchain-core==0.2.33
 langchain-google-vertexai==1.0.10

generative_ai/model_garden/test_model_garden_examples.py

@@ -31,9 +31,9 @@
 PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT")
 
 INPUT_BUCKET = "kellysun-test-project-europe-west1"
-OUTPUT_BUCKET = "python-docs-samples-tests"
+OUTPUT_BUCKET = "kellysun-test-project-europe-west1"
 OUTPUT_PATH = "batch/batch_text_predict_output"
-GCS_OUTPUT_PATH = "gs://python-docs-samples-tests/"
+GCS_OUTPUT_PATH = "gs://kellysun-test-project-europe-west1/"
 OUTPUT_TABLE = f"bq://{PROJECT_ID}.gen_ai_batch_prediction.predictions"
 
 
@@ -86,7 +86,7 @@ def test_generate_text() -> None:
 def test_batch_gemini_predict_gcs(output_folder: pytest.fixture()) -> None:
     output_uri = "gs://python-docs-samples-tests"
     job = _main_test(
-        test_func=lambda: claude_3_batch_prediction_gcs.batch_predict_gemini_createjob(
+        test_func=lambda: claude_3_batch_prediction_gcs.batch_predict_createjob(
             output_uri
         )
     )
@@ -96,7 +96,7 @@ def test_batch_gemini_predict_gcs(output_folder: pytest.fixture()) -> None:
 def test_batch_gemini_predict_bigquery(output_folder: pytest.fixture()) -> None:
     output_uri = f"bq://{PROJECT_ID}.gen_ai_batch_prediction.predictions"
     job = _main_test(
-        test_func=lambda: claude_3_batch_prediciton_bq.batch_predict_gemini_createjob(
+        test_func=lambda: claude_3_batch_prediciton_bq.batch_predict_createjob(
             output_uri
             )
     )