Merge pull request #272 from dchaley/issue271/fixContainerTag

Use container tag batch not gce
dchaley · Jul 9, 2024 · 40d3d57 · 40d3d57
2 parents 71cf41f + bf2f873
commit 40d3d57
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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"