kubeflow
/
pipelines
mirror of https://github.com/kubeflow/pipelines.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chore(components): Use new module for looking up ReFINED and AutoSxS image tags 

PiperOrigin-RevId: 605457014
This commit is contained in:
Michael Hu 2024-02-08 16:13:49 -08:00 committed by Google Cloud Pipeline Components maintainers
parent 21c5ffebb0
commit 1280753eb4
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
components/google-cloud/google_cloud_pipeline_components/_implementation/llm/env.py
View File

@ -12,17 +12,18 @@
# See the License for the specific language governing permissions and
# limitations under the License.
"""A collection of constants shared across components and pipelines."""
import os
_DEFAULT_AUTOSXS_IMAGE_TAG = '20240123_0507_RC00'
from google_cloud_pipeline_components._implementation.llm.generated import refined_image_versions
def get_private_image_tag() -> str:
return os.getenv('PRIVATE_IMAGE_TAG') or '20240124_0507_RC00'
return os.getenv('PRIVATE_IMAGE_TAG') or refined_image_versions.IMAGE_TAG
def get_autosxs_image_tag() -> str:
return os.getenv('PRIVATE_IMAGE_TAG') or _DEFAULT_AUTOSXS_IMAGE_TAG
return os.getenv('PRIVATE_IMAGE_TAG') or refined_image_versions.IMAGE_TAG
def get_use_test_machine_spec() -> bool: