Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

E2E image version using from environment variable instead of hardcoded one (#16968) 

* E2E image version using from environment variable instead of hardcoded one
This commit is contained in:
Dmitry Frolov 2023-03-06 12:06:11 -05:00 committed by GitHub
parent c2b28a0e8c
commit ae5ee1d4e2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/integrations_app/apps/custom_work_dependencies/app.py
View File

@ -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)