fix: LLM - Added missing parameters to the no-op `_TunableTextEmbeddi…

…ngModelMixin.get_tuned_model` method PiperOrigin-RevId: 630187156
googleapis · May 2, 2024 · eb05ac4 · eb05ac4
1 parent 78a95c5
commit eb05ac4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vertexai/language_models/_language_models.py b/vertexai/language_models/_language_models.py
@@ -2183,7 +2183,8 @@ async def get_embeddings_async(
 # task_type must be 'DEFAULT' or None if _model_id is textembedding-gecko@001.
 class _TunableTextEmbeddingModelMixin(_TunableModelMixin):
     @classmethod
-    def get_tuned_model():
+    def get_tuned_model(cls, *args, **kwargs):
+        del args, kwargs  # Unused.
         raise NotImplementedError(
             "Use deploy_tuned_model instead to get the tuned model."
         )