fix: Avoid unexpected keyword argument for sentence_transformers (#1269)

Now that remote-vllm include inline::sentence_transformers there is an
issue building the image:
Error building stack:
SentenceTransformersInferenceConfig.sample_run_config() got an
unexpected keyword argument '__distro_dir__'

To avoid that issue this fix extends the sample_run_config to accept
extra kwargs
This commit is contained in:
Luis Tomas Bolivar 2025-02-28 01:47:26 +01:00 committed by GitHub
parent c2d2a80b0a
commit 73c6f6126f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -11,5 +11,5 @@ from pydantic import BaseModel
class SentenceTransformersInferenceConfig(BaseModel): class SentenceTransformersInferenceConfig(BaseModel):
@classmethod @classmethod
def sample_run_config(cls) -> Dict[str, Any]: def sample_run_config(cls, **kwargs) -> Dict[str, Any]:
return {} return {}