feat(components): Update container URIs for embedding eval components

PiperOrigin-RevId: 559213412
Linchin · Aug 22, 2023 · f43272d · f43272d
1 parent eaec515
commit f43272d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 9 deletions.
diff --git a/...pipeline_components/_implementation/model_evaluation/llm_embedding_retrieval/component.py b/...pipeline_components/_implementation/model_evaluation/llm_embedding_retrieval/component.py
@@ -22,9 +22,7 @@
 from kfp.dsl import PIPELINE_ROOT_PLACEHOLDER
 
 
-_IMAGE_URI = 'gcr.io/model-evaluation-dev/llm_eval:navekshasood-test'
-# TODO(b/290838262): Update the image URI to point to
-# us-docker.pkg.dev/vertex-evaluation/public/llm:va.b before we launch.
+_IMAGE_URI = 'us-docker.pkg.dev/vertex-evaluation/public/llm:v0.3'
 
 
 @container_component

diff --git a/...ents/_implementation/model_evaluation/llm_information_retrieval_preprocessor/component.py b/...ents/_implementation/model_evaluation/llm_information_retrieval_preprocessor/component.py
@@ -20,9 +20,7 @@
 from kfp.dsl import PIPELINE_ROOT_PLACEHOLDER
 
 
-_IMAGE_URI = 'gcr.io/model-evaluation-dev/llm_eval:navekshasood-test'
-# TODO(b/290838262): Update the image URI to point to
-# us-docker.pkg.dev/vertex-evaluation/public/llm:va.b before we launch.
+_IMAGE_URI = 'us-docker.pkg.dev/vertex-evaluation/public/llm:v0.3'
 
 
 @container_component

diff --git a/...d_pipeline_components/_implementation/model_evaluation/llm_retrieval_metrics/component.py b/...d_pipeline_components/_implementation/model_evaluation/llm_retrieval_metrics/component.py
@@ -9,9 +9,7 @@
 from kfp.dsl import PIPELINE_ROOT_PLACEHOLDER
 
 
-_IMAGE_URI = 'gcr.io/model-evaluation-dev/llm_eval:navekshasood-test'
-# TODO(b/290838262): Update the image URI to point to
-# us-docker.pkg.dev/vertex-evaluation/public/llm:va.b before we launch.
+_IMAGE_URI = 'us-docker.pkg.dev/vertex-evaluation/public/llm:v0.3'
 
 
 @container_component