From ebc650bd95a4d09a8d77dc7e0379a6b9a909bb69 Mon Sep 17 00:00:00 2001 From: Sherin Thomas Date: Wed, 31 Aug 2022 18:57:21 +0530 Subject: [PATCH] E2E fix for custom base image (#14468) * new custom base image * image tag --- tests/tests_app_examples/custom_work_dependencies/app.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/tests_app_examples/custom_work_dependencies/app.py b/tests/tests_app_examples/custom_work_dependencies/app.py index fb85c91190..a7dbb7ff70 100644 --- a/tests/tests_app_examples/custom_work_dependencies/app.py +++ b/tests/tests_app_examples/custom_work_dependencies/app.py @@ -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):