[yaml] add RunInference support with VertexAI

Signed-off-by: Jeffrey Kinard <[email protected]>
apache · Dec 17, 2024 · c04712b · c04712b
1 parent 0e37501
commit c04712b
Show file tree

Hide file tree

Showing 6 changed files with 537 additions and 110 deletions.
diff --git a/sdks/python/apache_beam/yaml/standard_providers.yaml b/sdks/python/apache_beam/yaml/standard_providers.yaml
@@ -56,6 +56,7 @@
   config: {}
   transforms:
     MLTransform: 'apache_beam.yaml.yaml_ml.ml_transform'
+    RunInference: 'apache_beam.yaml.yaml_ml.run_inference'
 
 - type: renaming
   transforms: