Skip to content

Commit 0cc6784

Browse files
authored
docs: correct model deprecation warning for gemini text generator (#1562)
* docs: correct model deprecation warning for gemini text generator * add remove model warning * just remove unnecessary warning
1 parent 910be2b commit 0cc6784

File tree

2 files changed

+0
-19
lines changed

2 files changed

+0
-19
lines changed

bigframes/ml/llm.py

-5
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121

2222
import bigframes_vendored.constants as constants
2323
from google.cloud import bigquery
24-
import typing_extensions
2524

2625
from bigframes import dtypes, exceptions
2726
import bigframes.bigquery as bbq
@@ -403,10 +402,6 @@ def to_gbq(
403402
return new_model.session.read_gbq_model(model_name)
404403

405404

406-
@typing_extensions.deprecated(
407-
"gemini-1.5-X are going to be deprecated. Use gemini-2.0-X (https://cloud.google.com/python/docs/reference/bigframes/latest/bigframes.ml.llm.GeminiTextGenerator) instead. ",
408-
category=exceptions.ApiDeprecationWarning,
409-
)
410405
@log_adapter.class_logger
411406
class GeminiTextGenerator(base.RetriableRemotePredictor):
412407
"""Gemini text generator LLM model.

tests/system/small/ml/test_llm.py

-14
Original file line numberDiff line numberDiff line change
@@ -807,20 +807,6 @@ def test_llm_gemini_pro_score_params(llm_fine_tune_df_default_index, model_name)
807807
)
808808

809809

810-
@pytest.mark.parametrize(
811-
"model_name",
812-
(
813-
"gemini-1.5-pro-001",
814-
"gemini-1.5-pro-002",
815-
"gemini-1.5-flash-001",
816-
"gemini-1.5-flash-002",
817-
),
818-
)
819-
def test_gemini_text_generator_deprecated(model_name):
820-
with pytest.warns(exceptions.ApiDeprecationWarning):
821-
llm.GeminiTextGenerator(model_name=model_name)
822-
823-
824810
@pytest.mark.parametrize(
825811
"model_name",
826812
(

0 commit comments

Comments
 (0)