[YAML] fix renaming provider caching and YAML schema validation (#29290)

Signed-off-by: Jeffrey Kinard <[email protected]>
apache · Nov 3, 2023 · 058c947 · 058c947
1 parent 0e8e54c
commit 058c947
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/sdks/python/apache_beam/yaml/pipeline.schema.yaml b/sdks/python/apache_beam/yaml/pipeline.schema.yaml
@@ -159,7 +159,10 @@ properties:
       # These are the only top-level properties defined in pipeline.
       - type: object
         properties:
-          type: { const: chain }
+          type:
+            oneOf:
+              - { const: composite }
+              - { const: chain }
           windowing:
             $ref: '#/$defs/windowing'
           transforms: {}

diff --git a/sdks/python/apache_beam/yaml/yaml_provider.py b/sdks/python/apache_beam/yaml/yaml_provider.py
@@ -765,6 +765,9 @@ def _affinity(self, other):
   def underlying_provider(self):
     return self._underlying_provider.underlying_provider()
 
+  def cache_artifacts(self):
+    self._underlying_provider.cache_artifacts()
+
 
 def parse_providers(provider_specs):
   providers = collections.defaultdict(list)