fix(components): Update base image for KFP lightweight component for …

…VPC SC compliance PiperOrigin-RevId: 597352218
kubeflow · Jan 11, 2024 · ed54c46 · ed54c46
1 parent daa7299
commit ed54c46
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...line_components/_implementation/model_evaluation/llm_evaluation_preprocessor/component.py b/...line_components/_implementation/model_evaluation/llm_evaluation_preprocessor/component.py
@@ -22,7 +22,7 @@
 
 
 # pylint: disable=g-import-not-at-top, g-doc-args, unexpected-keyword-arg
-@dsl.component
+@dsl.component(base_image=version.LLM_EVAL_IMAGE_TAG)
 def add_json_escape_to_list(input_list: List[str]) -> str:
   import json