Add docker registry url

Yelp · Jan 9, 2025 · 29913ef · 29913ef
1 parent 013fcc5
commit 29913ef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/paasta_tools/tron_tools.py b/paasta_tools/tron_tools.py
@@ -357,7 +357,7 @@ def build_spark_config(self) -> Dict[str, str]:
             paasta_pool=self.get_spark_executor_pool(),
             paasta_service=self.get_service(),
             paasta_instance=self.get_instance(),
-            docker_img="$PAASTA_DOCKER_IMAGE",
+            docker_img=f"{self.get_docker_registry()}/$PAASTA_DOCKER_IMAGE",
             extra_volumes=self.get_volumes(
                 system_paasta_config.get_volumes(),
                 uses_bulkdata_default=system_paasta_config.get_uses_bulkdata_default(),

diff --git a/tests/test_tron_tools.py b/tests/test_tron_tools.py
@@ -1322,7 +1322,7 @@ def test_format_tron_action_dict_spark(
             "--conf spark.executorEnv.PAASTA_INSTANCE_TYPE=spark "
             "--conf spark.executorEnv.SPARK_EXECUTOR_DIRS=/tmp "
             "--conf spark.kubernetes.pyspark.pythonVersion=3 "
-            "--conf spark.kubernetes.container.image=$PAASTA_DOCKER_IMAGE "
+            "--conf spark.kubernetes.container.image=docker-registry.com:400/$PAASTA_DOCKER_IMAGE "
             "--conf spark.kubernetes.namespace=paasta-spark "
             "--conf spark.kubernetes.executor.label.yelp.com/paasta_service=my_service "
             "--conf spark.kubernetes.executor.label.yelp.com/paasta_instance=my_job.do_something "