attempt to fix test failure

open-ce · Jun 18, 2024 · 345adc3 · 345adc3
1 parent 2a342db
commit 345adc3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/open_ce/container_build.py b/open_ce/container_build.py
@@ -26,9 +26,9 @@
 from open_ce.inputs import Argument
 
 OPEN_CE_PATH = os.path.abspath(os.path.join(os.path.dirname(__file__), "."))
-BUILD_IMAGE_NAME = "builder"
+BUILD_IMAGE_NAME = "builder/ubi9"
 BUILD_IMAGE_PATH = os.path.join(OPEN_CE_PATH, "images", BUILD_IMAGE_NAME)
-BUILD_CUDA_IMAGE_NAME = "builder-cuda"
+BUILD_CUDA_IMAGE_NAME = "builder-cuda/ubi9"
 BUILD_CUDA_IMAGE_PATH = os.path.join(OPEN_CE_PATH, "images", BUILD_CUDA_IMAGE_NAME)
 LOCAL_FILES_PATH = os.path.join(os.path.join(os.getcwd(), "local_files"))