charlespnh opened a new issue, #35693:
URL: https://github.com/apache/beam/issues/35693
### What happened?
Run the following YAML pipeline:
```
pipeline:
type: chain
transforms:
- type: Create
name: CreateSampleData
config:
elements:
- { id: 1, log_message: "Error in module A" }
- { id: 2, log_message: "Warning in module B" }
- { id: 3, log_message: "Info in module C" }
- type: MLTransform
name: Preprocessing
config:
write_artifact_location: "gs://my-warehouse/beam_ml_artifacts"
transforms:
- type: SentenceTransformerEmbeddings
config: { model_name: all-MiniLM-L6-v2, columns: [ log_message ]
}
options:
yaml_experimental_features: ['ML']
```
... gives the following error:
```
ValueError: Error applying transform "Preprocessing" at line 12: Unknown ML
transform type: 'SentenceTransformerEmbeddings'
```
... but according to the doc
https://beam.apache.org/documentation/ml/preprocess-data/#use-mltransform,
`SentenceTransformerEmbeddings` is one of the available transformers to be used
with MLTransform
### Issue Priority
Priority: 2 (default / most bugs should be filed as P2)
### Issue Components
- [ ] Component: Python SDK
- [ ] Component: Java SDK
- [ ] Component: Go SDK
- [ ] Component: Typescript SDK
- [ ] Component: IO connector
- [x] Component: Beam YAML
- [ ] Component: Beam examples
- [ ] Component: Beam playground
- [ ] Component: Beam katas
- [ ] Component: Website
- [ ] Component: Infrastructure
- [ ] Component: Spark Runner
- [ ] Component: Flink Runner
- [ ] Component: Samza Runner
- [ ] Component: Twister2 Runner
- [ ] Component: Hazelcast Jet Runner
- [ ] Component: Google Cloud Dataflow Runner
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
