From a8a4c38bd75ed62064770faf5685856acfe5cdb3 Mon Sep 17 00:00:00 2001 From: Eric Schmidt Date: Fri, 20 Sep 2024 14:16:21 -0700 Subject: [PATCH] lint --- generative_ai/gemma2_predict_gpu_test.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/generative_ai/gemma2_predict_gpu_test.py b/generative_ai/gemma2_predict_gpu_test.py index d335d5bd33db..94ab92b60eb5 100644 --- a/generative_ai/gemma2_predict_gpu_test.py +++ b/generative_ai/gemma2_predict_gpu_test.py @@ -18,8 +18,7 @@ from google.cloud.aiplatform import models # Interfaces defined here: -# https://github.com/googleapis/googleapis/tree/master/google/cloud/aiplatform/v1 -# https://github.com/googleapis/googleapis-gen/tree/master/google/cloud/aiplatform/v1/aiplatform-v1-py +# https://github.com/googleapis/python-aiplatform/blob/main/google/cloud/aiplatform/models.py from gemma2_predict_gpu import endpoint_predict_sample @@ -52,8 +51,7 @@ def test_gemma2_predict_gpu(mock_init, mock_prediction, mock_endpoint): endpoint = "fake_endpoint_id" prediction = endpoint_predict_sample( - project=project_id, location=location, instances=[instance], - endpoint=endpoint + project=project_id, location=location, instances=[instance], endpoint=endpoint ) assert prediction