diff --git a/sdks/python/apache_beam/ml/transforms/embeddings/huggingface.py b/sdks/python/apache_beam/ml/transforms/embeddings/huggingface.py index c3eeccf2a0f0..56a9535b5764 100644 --- a/sdks/python/apache_beam/ml/transforms/embeddings/huggingface.py +++ b/sdks/python/apache_beam/ml/transforms/embeddings/huggingface.py @@ -210,12 +210,11 @@ def load_model(self): def run_inference( self, batch, session: requests.Session, inference_args=None): + # HF router feature-extraction accepts inputs only (not nested options). response = session.post( self._config.api_url, headers=self._config.authorization_token, - json={ - "inputs": batch, "options": inference_args - }) + json={"inputs": batch}) return response.json()