From bf2f873f6fed3b8cf8825cd688863766b7b60867 Mon Sep 17 00:00:00 2001 From: David Haley Date: Mon, 8 Jul 2024 17:01:12 -0700 Subject: [PATCH] Use container tag batch not gce --- batch/run-job.py | 2 +- batch/run-multistep-job.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/batch/run-job.py b/batch/run-job.py index dfd1254..5d060ca 100755 --- a/batch/run-job.py +++ b/batch/run-job.py @@ -27,7 +27,7 @@ model_path = args.model_path -CONTAINER_IMAGE = "us-central1-docker.pkg.dev/deepcell-on-batch/deepcell-benchmarking-us-central1/benchmarking:gce" +CONTAINER_IMAGE = "us-central1-docker.pkg.dev/deepcell-on-batch/deepcell-benchmarking-us-central1/benchmarking:batch" OUTPUT_BASE_PATH = "gs://deepcell-batch-jobs_us-central1/job-runs" REGION = "us-central1" diff --git a/batch/run-multistep-job.py b/batch/run-multistep-job.py index 09e51b8..dfb5d3f 100755 --- a/batch/run-multistep-job.py +++ b/batch/run-multistep-job.py @@ -42,7 +42,7 @@ input_channels_path = args.input_channels_path bigquery_benchmarking_table = args.bigquery_benchmarking_table -CONTAINER_IMAGE = "us-central1-docker.pkg.dev/deepcell-on-batch/deepcell-benchmarking-us-central1/benchmarking:gce" +CONTAINER_IMAGE = "us-central1-docker.pkg.dev/deepcell-on-batch/deepcell-benchmarking-us-central1/benchmarking:batch" OUTPUT_BASE_PATH = "gs://deepcell-batch-jobs_us-central1/job-runs" REGION = "us-central1"