This is an automated email from the ASF dual-hosted git repository.
derrickaw pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git
The following commit(s) were added to refs/heads/master by this push:
new b60082d17d4 Revert "[yaml] - Add huggingface model handler (#38110)"
(#38421)
b60082d17d4 is described below
commit b60082d17d4c86a6cdd98e23778490aceb40b432
Author: Derrick Williams <[email protected]>
AuthorDate: Fri May 8 18:04:28 2026 -0400
Revert "[yaml] - Add huggingface model handler (#38110)" (#38421)
This reverts commit 422f6304af199207462127d09d58fddc33398052.
---
.../beam_PostCommit_Yaml_Xlang_Direct.yml | 2 +-
.../workflows/beam_PreCommit_Yaml_Xlang_Direct.yml | 2 +-
...uninference_vertexai.yaml => runinference.yaml} | 0
.../yaml/tests/runinference_huggingface.yaml | 62 ----------------------
sdks/python/apache_beam/yaml/yaml_ml.py | 49 -----------------
sdks/python/build.gradle | 14 +----
6 files changed, 4 insertions(+), 125 deletions(-)
diff --git a/.github/workflows/beam_PostCommit_Yaml_Xlang_Direct.yml
b/.github/workflows/beam_PostCommit_Yaml_Xlang_Direct.yml
index afa437b64de..ea1c255f7cc 100644
--- a/.github/workflows/beam_PostCommit_Yaml_Xlang_Direct.yml
+++ b/.github/workflows/beam_PostCommit_Yaml_Xlang_Direct.yml
@@ -80,7 +80,7 @@ jobs:
- name: run PostCommit Yaml Xlang Direct script
uses: ./.github/actions/gradle-command-self-hosted-action
with:
- gradle-command: :sdks:python:postCommitYamlIntegrationTests
-PyamlTestSet=${{ matrix.test_set }}
+ gradle-command: :sdks:python:postCommitYamlIntegrationTests
-PyamlTestSet=${{ matrix.test_set }} -PbeamPythonExtra=ml_test,yaml
- name: Archive Python Test Results
uses: actions/upload-artifact@v7
if: failure()
diff --git a/.github/workflows/beam_PreCommit_Yaml_Xlang_Direct.yml
b/.github/workflows/beam_PreCommit_Yaml_Xlang_Direct.yml
index 0b8f4cd6393..7d17fd2140c 100644
--- a/.github/workflows/beam_PreCommit_Yaml_Xlang_Direct.yml
+++ b/.github/workflows/beam_PreCommit_Yaml_Xlang_Direct.yml
@@ -91,7 +91,7 @@ jobs:
- name: run PreCommit Yaml Xlang Direct script
uses: ./.github/actions/gradle-command-self-hosted-action
with:
- gradle-command: :sdks:python:yamlIntegrationTests
+ gradle-command: :sdks:python:yamlIntegrationTests
-PbeamPythonExtra=ml_test,yaml
- name: Archive Python Test Results
uses: actions/upload-artifact@v7
if: failure()
diff --git a/sdks/python/apache_beam/yaml/tests/runinference_vertexai.yaml
b/sdks/python/apache_beam/yaml/tests/runinference.yaml
similarity index 100%
rename from sdks/python/apache_beam/yaml/tests/runinference_vertexai.yaml
rename to sdks/python/apache_beam/yaml/tests/runinference.yaml
diff --git a/sdks/python/apache_beam/yaml/tests/runinference_huggingface.yaml
b/sdks/python/apache_beam/yaml/tests/runinference_huggingface.yaml
deleted file mode 100644
index 8728a6f544a..00000000000
--- a/sdks/python/apache_beam/yaml/tests/runinference_huggingface.yaml
+++ /dev/null
@@ -1,62 +0,0 @@
-# Licensed to the Apache Software Foundation (ASF) under one or more
-# contributor license agreements. See the NOTICE file distributed with
-# this work for additional information regarding copyright ownership.
-# The ASF licenses this file to You under the Apache License, Version 2.0
-# (the "License"); you may not use this file except in compliance with
-# the License. You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-pipelines:
- - pipeline:
- type: chain
- transforms:
- - type: Create
- config:
- elements:
- - text: "I love Apache Beam!"
- - text: "I hate this error."
- - type: RunInference
- config:
- model_handler:
- type: "HuggingFacePipeline"
- config:
- task: "text-classification"
- inference_fn:
- callable: |
- def real_inference(batch, pipeline, inference_args):
- predictions = pipeline(batch, **inference_args)
-
- # If it's a single dictionary (batch size of 1), wrap it
in a list
- if isinstance(predictions, dict):
- predictions = [predictions]
-
- return {
- 'label': [p['label'] for p in predictions],
- 'score': [p['score'] for p in predictions]
- }
- preprocess:
- callable: 'lambda x: x.text'
- - type: MapToFields
- config:
- language: python
- fields:
- text: text
- sentiment:
- callable: 'lambda x: x.inference.inference["label"]'
- - type: AssertEqual
- config:
- elements:
- - text: "I love Apache Beam!"
- sentiment: "POSITIVE"
- - text: "I hate this error."
- sentiment: "NEGATIVE"
-
- options:
- yaml_experimental_features: ['ML']
diff --git a/sdks/python/apache_beam/yaml/yaml_ml.py
b/sdks/python/apache_beam/yaml/yaml_ml.py
index 05cbed3bd45..51f18c73304 100644
--- a/sdks/python/apache_beam/yaml/yaml_ml.py
+++ b/sdks/python/apache_beam/yaml/yaml_ml.py
@@ -282,55 +282,6 @@ class
VertexAIModelHandlerJSONProvider(ModelHandlerProvider):
('model_id', Optional[str])])
[email protected]_handler_type('HuggingFacePipeline')
-class HuggingFacePipelineProvider(ModelHandlerProvider):
- def __init__(
- self,
- task: Optional[str] = None,
- model: Optional[str] = None,
- preprocess: Optional[dict[str, str]] = None,
- postprocess: Optional[dict[str, str]] = None,
- device: Optional[Any] = None,
- inference_fn: Optional[dict[str, str]] = None,
- load_pipeline_args: Optional[dict[str, Any]] = None,
- **kwargs):
- try:
- from apache_beam.ml.inference.huggingface_inference import
HuggingFacePipelineModelHandler
- except ImportError:
- raise ValueError(
- 'Unable to import HuggingFacePipelineModelHandler. Please '
- 'install transformers dependencies.')
-
- kwargs = {k: v for k, v in kwargs.items() if not k.startswith('_')}
-
- inference_fn_obj = self.parse_processing_transform(
- inference_fn, 'inference_fn') if inference_fn else None
-
- handler_kwargs = {}
- if inference_fn_obj:
- handler_kwargs['inference_fn'] = inference_fn_obj
-
- _handler = HuggingFacePipelineModelHandler(
- task=task,
- model=model,
- device=device,
- load_pipeline_args=load_pipeline_args,
- **handler_kwargs,
- **kwargs)
-
- super().__init__(_handler, preprocess, postprocess)
-
- @staticmethod
- def validate(config):
- if not config.get('task') and not config.get('model'):
- raise ValueError(
- "HuggingFacePipeline requires either 'task' or "
- "'model' to be specified.")
-
- def inference_output_type(self):
- return Any
-
-
@beam.ptransform.ptransform_fn
def run_inference(
pcoll,
diff --git a/sdks/python/build.gradle b/sdks/python/build.gradle
index e676fd11043..5f09dff57e8 100644
--- a/sdks/python/build.gradle
+++ b/sdks/python/build.gradle
@@ -124,20 +124,10 @@ tasks.register("generateYamlDocs") {
outputs.file "${buildDir}/yaml-examples.html"
}
-tasks.register("installYamlIntegrationTestDeps") {
- dependsOn installGcpTest
- doLast {
- exec {
- executable 'sh'
- args '-c', ". ${envdir}/bin/activate && pip install --pre --retries 10
${buildDir}/apache-beam.tar.gz[ml_test,yaml,transformers]"
- }
- }
-}
-
tasks.register("yamlIntegrationTests") {
description "Runs precommit integration tests for yaml pipelines."
- dependsOn installYamlIntegrationTestDeps
+ dependsOn installGcpTest
// Need to build all expansion services referenced in apache_beam/yaml/*.*
// grep -oh 'sdk.*Jar' sdks/python/apache_beam/yaml/*.yaml | sort | uniq
dependsOn ":sdks:java:extensions:schemaio-expansion-service:shadowJar"
@@ -156,7 +146,7 @@ tasks.register("yamlIntegrationTests") {
tasks.register("postCommitYamlIntegrationTests") {
description "Runs postcommit integration tests for yaml pipelines -
parameterized by yamlTestSet."
- dependsOn installYamlIntegrationTestDeps
+ dependsOn installGcpTest
// Need to build all expansion services referenced in apache_beam/yaml/*.*
// grep -oh 'sdk.*Jar' sdks/python/apache_beam/yaml/*.yaml | sort | uniq
dependsOn ":sdks:java:extensions:schemaio-expansion-service:shadowJar"
