[yaml] add RunInference support with VertexAI (#33406)

* [yaml] add RunInference support with VertexAI Signed-off-by: Jeffrey Kinard <jeff@thekinards.com> * address comments and fix tests Signed-off-by: Jeffrey Kinard <jeff@thekinards.com> * add more docs Signed-off-by: Jeffrey Kinard <jeff@thekinards.com> * fix failing tests Signed-off-by: Jeffrey Kinard <jeff@thekinards.com> * fix errors Signed-off-by: Jeffrey Kinard <jeff@thekinards.com> * fix lint Signed-off-by: Jeffrey Kinard <jeff@thekinards.com> --------- Signed-off-by: Jeffrey Kinard <jeff@thekinards.com>
apache · Dec 26, 2024 · 7e077dc · 7e077dc
1 parent 7c86bf3
commit 7e077dc
Show file tree

Hide file tree

Showing 6 changed files with 592 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: