Skip to content

Commit 0a80470

Browse files
committed
Add project argument
1 parent a3dafe4 commit 0a80470

3 files changed

Lines changed: 6 additions & 0 deletions

File tree

sdks/python/apache_beam/examples/inference/pytorch_image_captioning.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -341,6 +341,8 @@ def parse_known_args(argv):
341341

342342
# I/O & runtime
343343
parser.add_argument('--mode', default='batch', choices=['batch'])
344+
parser.add_argument(
345+
'--project', default='apache-beam-testing', help='GCP project ID')
344346
parser.add_argument(
345347
'--input', required=True, help='GCS path to file with image URIs')
346348
parser.add_argument(

sdks/python/apache_beam/examples/inference/pytorch_image_object_detection.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -243,6 +243,8 @@ def parse_known_args(argv):
243243

244244
# I/O & runtime
245245
parser.add_argument('--mode', default='batch', choices=['batch'])
246+
parser.add_argument(
247+
'--project', default='apache-beam-testing', help='GCP project ID')
246248
parser.add_argument(
247249
'--output_table',
248250
required=True,

sdks/python/apache_beam/examples/inference/pytorch_imagenet_rightfit.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -202,6 +202,8 @@ def process(self, kv: Tuple[str, PredictionResult]):
202202
def parse_known_args(argv):
203203
parser = argparse.ArgumentParser()
204204
# I/O & runtime
205+
parser.add_argument(
206+
'--project', default='apache-beam-testing', help='GCP project ID')
205207
parser.add_argument(
206208
'--mode', default='streaming', choices=['streaming', 'batch'])
207209
parser.add_argument(

0 commit comments

Comments
 (0)