From ae5ee1d4e29e415b61a5cf3daf55f72884e3e891 Mon Sep 17 00:00:00 2001 From: Dmitry Frolov Date: Mon, 6 Mar 2023 12:06:11 -0500 Subject: [PATCH] E2E image version using from environment variable instead of hardcoded one (#16968) * E2E image version using from environment variable instead of hardcoded one --- tests/integrations_app/apps/custom_work_dependencies/app.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/integrations_app/apps/custom_work_dependencies/app.py b/tests/integrations_app/apps/custom_work_dependencies/app.py index d09d3048b3..10f9992b9a 100644 --- a/tests/integrations_app/apps/custom_work_dependencies/app.py +++ b/tests/integrations_app/apps/custom_work_dependencies/app.py @@ -24,8 +24,7 @@ class WorkWithCustomBaseImage(LightningWork): def __init__(self, cloud_compute: CloudCompute = CloudCompute(), **kwargs): # this image has been created from ghcr.io/gridai/base-images:v1.8-cpu # by just adding an empty file at /content/.e2e_test - image_tag = "v1.15" - # image_tag = os.getenv("LIGHTNING_E2E_TEST_IMAGE_VERSION", "v1.15") + image_tag = os.getenv("LIGHTNING_E2E_TEST_IMAGE_VERSION", "v1.15") custom_image = f"ghcr.io/gridai/image-for-testing-custom-images-in-e2e:{image_tag}" build_config = BuildConfig(image=custom_image) super().__init__(parallel=True, **kwargs, cloud_compute=cloud_compute, cloud_build_config=build_config)