E2E fix for custom base image (#14468)
* new custom base image * image tag
This commit is contained in:
parent
00aefa82b7
commit
ebc650bd95
tests/tests_app_examples/custom_work_dependencies
|
@ -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):
|
||||
|
|
Loading…
Reference in New Issue