diff --git a/tests/integration-tests/tests/createami/test_createami.py b/tests/integration-tests/tests/createami/test_createami.py index 1145494cc2..4239fd62ae 100644 --- a/tests/integration-tests/tests/createami/test_createami.py +++ b/tests/integration-tests/tests/createami/test_createami.py @@ -157,6 +157,10 @@ def test_build_image( enable_lustre_client = False if os in ["alinux2", "alinux2023", "rocky9"]: update_os_packages = True + + # Disable DCV installation for Ubuntu 24.04 to avoid build failures with DLAMI + enable_dcv = os != "ubuntu2404" + image_config = pcluster_config_reader( config_file="image.config.yaml", parent_image=base_ami, @@ -165,6 +169,7 @@ def test_build_image( enable_nvidia=str(enable_nvidia and get_gpu_count(instance) > 0).lower(), update_os_packages=str(update_os_packages).lower(), enable_lustre_client=str(enable_lustre_client).lower(), + enable_dcv=str(enable_dcv).lower(), ) image = images_factory(image_id, image_config, region) diff --git a/tests/integration-tests/tests/createami/test_createami/test_build_image/image.config.yaml b/tests/integration-tests/tests/createami/test_createami/test_build_image/image.config.yaml index db15a8ece3..2a3a8bd9b1 100644 --- a/tests/integration-tests/tests/createami/test_createami/test_build_image/image.config.yaml +++ b/tests/integration-tests/tests/createami/test_createami/test_build_image/image.config.yaml @@ -36,3 +36,8 @@ DeploymentSettings: - {{ default_vpc_security_group_id }} DevSettings: TerminateInstanceOnFailure: True +{% if enable_dcv == "false" %} + Cookbook: + ExtraChefAttributes: | + {"cluster": {"dcv": {"install_enabled": false}}} +{% endif %}