From 328c69d739b4b6ebf87f5b4951bdb130c7fc9b6d Mon Sep 17 00:00:00 2001 From: sirutBuasai Date: Tue, 10 Feb 2026 13:20:14 -0800 Subject: [PATCH 1/4] pin setuptools during safety test --- dlc_developer_config.toml | 8 ++++---- src/utils.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/dlc_developer_config.toml b/dlc_developer_config.toml index 207784d1f191..fab97a2a7777 100644 --- a/dlc_developer_config.toml +++ b/dlc_developer_config.toml @@ -37,12 +37,12 @@ deep_canary_mode = false [build] # Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image. # available frameworks - ["base", "vllm", "sglang", "autogluon", "huggingface_vllm", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"] -build_frameworks = [] +build_frameworks = ["pytorch"] # By default we build both training and inference containers. Set true/false values to determine which to build. build_training = true -build_inference = true +build_inference = false # Set do_build to "false" to skip builds and test the latest image built by this PR # Note: at least one build is required to set do_build to "false" @@ -124,7 +124,7 @@ nightly_pr_test_mode = false dlc-pr-base = "" # Standard Framework Training -dlc-pr-pytorch-training = "" +dlc-pr-pytorch-training = "pytorch/training/buildspec-2-8-ec2.yml" dlc-pr-tensorflow-2-training = "" dlc-pr-autogluon-training = "" @@ -190,4 +190,4 @@ dlc-pr-vllm = "" dlc-pr-huggingface-vllm = "" # sglang -dlc-pr-sglang = "" \ No newline at end of file +dlc-pr-sglang = "" diff --git a/src/utils.py b/src/utils.py index c913c3a43b0d..0fb194ec025e 100644 --- a/src/utils.py +++ b/src/utils.py @@ -423,7 +423,7 @@ def generate_safety_report_for_image(image_uri, image_info, storage_file_path=No ctx = Context() docker_run_cmd = f"docker run -id --entrypoint='/bin/bash' {image_uri} " container_id = ctx.run(f"{docker_run_cmd}", hide=True, warn=True).stdout.strip() - install_safety_cmd = "pip install 'safety>=2.2.0,<3'" + install_safety_cmd = "pip install 'setuptools<82' 'safety>=2.2.0,<3'" docker_exec_cmd = f"docker exec -i {container_id}" ctx.run(f"{docker_exec_cmd} {install_safety_cmd}", hide=True, warn=True) ignore_dict = get_safety_ignore_dict( From b143657e04804661f9d78b4b29bdfc829f3c6a46 Mon Sep 17 00:00:00 2001 From: sirutBuasai Date: Tue, 10 Feb 2026 15:34:24 -0800 Subject: [PATCH 2/4] revert toml --- dlc_developer_config.toml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/dlc_developer_config.toml b/dlc_developer_config.toml index fab97a2a7777..207784d1f191 100644 --- a/dlc_developer_config.toml +++ b/dlc_developer_config.toml @@ -37,12 +37,12 @@ deep_canary_mode = false [build] # Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image. # available frameworks - ["base", "vllm", "sglang", "autogluon", "huggingface_vllm", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"] -build_frameworks = ["pytorch"] +build_frameworks = [] # By default we build both training and inference containers. Set true/false values to determine which to build. build_training = true -build_inference = false +build_inference = true # Set do_build to "false" to skip builds and test the latest image built by this PR # Note: at least one build is required to set do_build to "false" @@ -124,7 +124,7 @@ nightly_pr_test_mode = false dlc-pr-base = "" # Standard Framework Training -dlc-pr-pytorch-training = "pytorch/training/buildspec-2-8-ec2.yml" +dlc-pr-pytorch-training = "" dlc-pr-tensorflow-2-training = "" dlc-pr-autogluon-training = "" @@ -190,4 +190,4 @@ dlc-pr-vllm = "" dlc-pr-huggingface-vllm = "" # sglang -dlc-pr-sglang = "" +dlc-pr-sglang = "" \ No newline at end of file From ef73d13aa09bf6dfd5445f6128acb5f40771b603 Mon Sep 17 00:00:00 2001 From: sirutBuasai Date: Tue, 10 Feb 2026 16:00:16 -0800 Subject: [PATCH 3/4] run safety check --- dlc_developer_config.toml | 10 +++++----- test/dlc_tests/sanity/test_safety_check.py | 2 +- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/dlc_developer_config.toml b/dlc_developer_config.toml index 207784d1f191..267a160691fc 100644 --- a/dlc_developer_config.toml +++ b/dlc_developer_config.toml @@ -37,16 +37,16 @@ deep_canary_mode = false [build] # Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image. # available frameworks - ["base", "vllm", "sglang", "autogluon", "huggingface_vllm", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"] -build_frameworks = [] +build_frameworks = ["pytorch"] # By default we build both training and inference containers. Set true/false values to determine which to build. build_training = true -build_inference = true +build_inference = false # Set do_build to "false" to skip builds and test the latest image built by this PR # Note: at least one build is required to set do_build to "false" -do_build = true +do_build = false [notify] ### Notify on test failures @@ -124,7 +124,7 @@ nightly_pr_test_mode = false dlc-pr-base = "" # Standard Framework Training -dlc-pr-pytorch-training = "" +dlc-pr-pytorch-training = "pytorch/training/buildspec-2-8-ec2.yml" dlc-pr-tensorflow-2-training = "" dlc-pr-autogluon-training = "" @@ -190,4 +190,4 @@ dlc-pr-vllm = "" dlc-pr-huggingface-vllm = "" # sglang -dlc-pr-sglang = "" \ No newline at end of file +dlc-pr-sglang = "" diff --git a/test/dlc_tests/sanity/test_safety_check.py b/test/dlc_tests/sanity/test_safety_check.py index 0303529fc6ac..1aa4d94d156f 100644 --- a/test/dlc_tests/sanity/test_safety_check.py +++ b/test/dlc_tests/sanity/test_safety_check.py @@ -1094,7 +1094,7 @@ def test_safety(image): hide=True, ) try: - run(f"{docker_exec_cmd} pip install 'safety>=2.2.0' yolk3k ", hide=True) + run(f"{docker_exec_cmd} pip install 'setuptools<82' 'safety>=2.2.0' yolk3k ", hide=True) json_str_safety_result = extract_json_from_safety_output( safety_check.run_safety_check_on_container(docker_exec_cmd) ) From de72db80a616fa8e9d2c037d852663535b511d3a Mon Sep 17 00:00:00 2001 From: sirutBuasai Date: Tue, 10 Feb 2026 16:02:00 -0800 Subject: [PATCH 4/4] revert toml --- dlc_developer_config.toml | 10 +++++----- test/dlc_tests/sanity/test_safety_check.py | 2 +- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/dlc_developer_config.toml b/dlc_developer_config.toml index 267a160691fc..207784d1f191 100644 --- a/dlc_developer_config.toml +++ b/dlc_developer_config.toml @@ -37,16 +37,16 @@ deep_canary_mode = false [build] # Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image. # available frameworks - ["base", "vllm", "sglang", "autogluon", "huggingface_vllm", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"] -build_frameworks = ["pytorch"] +build_frameworks = [] # By default we build both training and inference containers. Set true/false values to determine which to build. build_training = true -build_inference = false +build_inference = true # Set do_build to "false" to skip builds and test the latest image built by this PR # Note: at least one build is required to set do_build to "false" -do_build = false +do_build = true [notify] ### Notify on test failures @@ -124,7 +124,7 @@ nightly_pr_test_mode = false dlc-pr-base = "" # Standard Framework Training -dlc-pr-pytorch-training = "pytorch/training/buildspec-2-8-ec2.yml" +dlc-pr-pytorch-training = "" dlc-pr-tensorflow-2-training = "" dlc-pr-autogluon-training = "" @@ -190,4 +190,4 @@ dlc-pr-vllm = "" dlc-pr-huggingface-vllm = "" # sglang -dlc-pr-sglang = "" +dlc-pr-sglang = "" \ No newline at end of file diff --git a/test/dlc_tests/sanity/test_safety_check.py b/test/dlc_tests/sanity/test_safety_check.py index 1aa4d94d156f..0303529fc6ac 100644 --- a/test/dlc_tests/sanity/test_safety_check.py +++ b/test/dlc_tests/sanity/test_safety_check.py @@ -1094,7 +1094,7 @@ def test_safety(image): hide=True, ) try: - run(f"{docker_exec_cmd} pip install 'setuptools<82' 'safety>=2.2.0' yolk3k ", hide=True) + run(f"{docker_exec_cmd} pip install 'safety>=2.2.0' yolk3k ", hide=True) json_str_safety_result = extract_json_from_safety_output( safety_check.run_safety_check_on_container(docker_exec_cmd) )