E2E fix for custom base image ()

* new custom base image

* image tag
This commit is contained in:
Sherin Thomas 2022-08-31 18:57:21 +05:30 committed by GitHub
parent 00aefa82b7
commit ebc650bd95
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
tests/tests_app_examples/custom_work_dependencies

View File

@ -28,13 +28,13 @@ 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
custom_image = "ghcr.io/gridai/image-for-testing-custom-images-in-e2e"
custom_image = "ghcr.io/gridai/image-for-testing-custom-images-in-e2e:v0.0.1"
build_config = BuildConfig(image=custom_image)
super().__init__(parallel=True, **kwargs, cloud_compute=cloud_compute, cloud_build_config=build_config)
def run(self):
# checking the existence of the file - this file had been added to the custom base image
assert ".e2e_test" in os.listdir("/content/"), "file not found"
assert ".e2e_test" in os.listdir("/testdir/"), "file not found"
class CustomWorkBuildConfigChecker(LightningFlow):