Skip to content

Commit

Permalink
[BEAM-12447] Upgrade cloud build client and add/cleanup options (apac…
Browse files Browse the repository at this point in the history
  • Loading branch information
y1chi authored and nancyxu123 committed Mar 9, 2022
1 parent ac568c7 commit ed4e751
Show file tree
Hide file tree
Showing 6 changed files with 3,387 additions and 361 deletions.
9 changes: 3 additions & 6 deletions sdks/python/apache_beam/options/pipeline_options.py
Original file line number Diff line number Diff line change
Expand Up @@ -1176,14 +1176,11 @@ def _add_argparse_args(cls, parser):
'can also subclass SdkContainerImageBuilder and use that to build '
'in other environments.'))
parser.add_argument(
'--prebuild_sdk_container_base_image',
'--cloud_build_machine_type',
default=None,
help=(
'The base image to use when pre-building the sdk container image '
'with dependencies, if not specified, by default the released '
'public apache beam python sdk container image corresponding to '
'the sdk version will be used, if a dev sdk is used the base '
'image will default to the latest released sdk image.'))
'If specified, use the machine type explicitly when prebuilding'
'SDK container image on Google Cloud Build.'))
parser.add_argument(
'--docker_registry_push_url',
default=None,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,11 @@
Dataflow client utility functions."""

# pytype: skip-file
# To regenerate the client:
# pip install google-apitools[cli]
# gen_client --discovery_url=cloudbuild.v1 --overwrite \
# --outdir=apache_beam/runners/dataflow/internal/clients/cloudbuild \
# --root_package=. client

import codecs
from functools import partial
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,11 +25,9 @@
# pylint: disable=wrong-import-order, wrong-import-position
try:
from apitools.base.py import *

from apache_beam.runners.dataflow.internal.clients.cloudbuild.cloudbuild_v1_client import *
from apache_beam.runners.dataflow.internal.clients.cloudbuild.cloudbuild_v1_messages import *
except ImportError:
pass
# pylint: enable=wrong-import-order, wrong-import-position

__path__ = pkgutil.extend_path(__path__, __name__)
Loading

0 comments on commit ed4e751

Please sign in to comment.