Skip to content

Commit eb05ac4

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
fix: LLM - Added missing parameters to the no-op _TunableTextEmbeddingModelMixin.get_tuned_model method
PiperOrigin-RevId: 630187156
1 parent 78a95c5 commit eb05ac4

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vertexai/language_models/_language_models.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -2183,7 +2183,8 @@ async def get_embeddings_async(
21832183
# task_type must be 'DEFAULT' or None if _model_id is textembedding-gecko@001.
21842184
class _TunableTextEmbeddingModelMixin(_TunableModelMixin):
21852185
@classmethod
2186-
def get_tuned_model():
2186+
def get_tuned_model(cls, *args, **kwargs):
2187+
del args, kwargs # Unused.
21872188
raise NotImplementedError(
21882189
"Use deploy_tuned_model instead to get the tuned model."
21892190
)

0 commit comments

Comments
 (0)