Skip to content

Commit 02d53af

Browse files
authored
Merge pull request #34404 Factor out a get_default_sql_provider() function.
2 parents 5f583ea + 3a4a0f7 commit 02d53af

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

sdks/python/apache_beam/yaml/yaml_provider.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -661,16 +661,20 @@ def create_transform(self, type, args, yaml_create_transform):
661661
return self._transform_factories[type](yaml_create_transform, **args)
662662

663663

664+
def get_default_sql_provider():
665+
return beam_jar(
666+
urns={'Sql': 'beam:external:java:sql:v1'},
667+
gradle_target='sdks:java:extensions:sql:expansion-service:shadowJar')
668+
669+
664670
class SqlBackedProvider(Provider):
665671
def __init__(
666672
self,
667673
transforms: Mapping[str, Callable[..., beam.PTransform]],
668674
sql_provider: Optional[Provider] = None):
669675
self._transforms = transforms
670676
if sql_provider is None:
671-
sql_provider = beam_jar(
672-
urns={'Sql': 'beam:external:java:sql:v1'},
673-
gradle_target='sdks:java:extensions:sql:expansion-service:shadowJar')
677+
sql_provider = get_default_sql_provider()
674678
self._sql_provider = sql_provider
675679

676680
def sql_provider(self):

0 commit comments

Comments
 (0)