update black formatter to 24.8.0 (#4952)

* update black formatter

* reformatted 25.1.0

* removed 25.1.0

* downgrade to 24.8

* downgrade to 24.8

Author: Jyothirmaikottu

.github/workflows/black.yml

@@ -7,6 +7,6 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v3
-      - uses: psf/black@23.12.1
+      - uses: psf/black@24.8.0
         with:
           options: "--check --verbose -l 100"

scheduler/job_requester/requester.py

@@ -116,9 +116,7 @@ def assign_sagemaker_instance_type(self, image):
         return (
             "ml.g5.12xlarge"
             if "gpu" in image
-            else "ml.c5.4xlarge"
-            if "tensorflow" in image
-            else "ml.c5.9xlarge"
+            else "ml.c5.4xlarge" if "tensorflow" in image else "ml.c5.9xlarge"
         )
 
     def extract_timestamp(self, ticket_key):

src/patch_helper.py

@@ -216,13 +216,13 @@ def conduct_autopatch_build_setup(pre_push_image_object: DockerImage, download_p
             s3_downloaded_path=download_path,
             python_version=info.get("python_version"),
         )
-        THREADS[
-            f"trigger_enhanced_scan_patching-{base_image_uri_for_patch_builds}"
-        ] = executor.submit(
-            trigger_enhanced_scan_patching,
-            image_uri=base_image_uri_for_patch_builds,
-            patch_details_path=current_patch_details_path,
-            python_version=info.get("python_version"),
+        THREADS[f"trigger_enhanced_scan_patching-{base_image_uri_for_patch_builds}"] = (
+            executor.submit(
+                trigger_enhanced_scan_patching,
+                image_uri=base_image_uri_for_patch_builds,
+                patch_details_path=current_patch_details_path,
+                python_version=info.get("python_version"),
+            )
         )
     FORMATTER.progress(THREADS)
 

src/prepare_dlc_dev_environment.py

@@ -599,9 +599,9 @@ def handle_restore_option(toml_path, buildspec_paths, to_commit, to_push):
     if buildspec_paths:
         for buildspec_path in buildspec_paths:
             orig_buildspec = restore_buildspec(buildspec_path)
-            changes[
-                os.path.join(get_cloned_folder_path(), buildspec_path)
-            ] = f"Restored {buildspec_path} to {orig_buildspec}"
+            changes[os.path.join(get_cloned_folder_path(), buildspec_path)] = (
+                f"Restored {buildspec_path} to {orig_buildspec}"
+            )
 
     if not buildspec_paths or toml_path not in changes:
         restore_default_toml(toml_path)

src/requirements.txt

@@ -7,7 +7,7 @@ docker==6.1.3
 pyfiglet==0.8.post1
 reprint==0.5.2
 ruamel.yaml==0.18.10
-black==23.3.0
+black==24.8.0
 junit-xml==1.9
 toml==0.10.2
 retrying

src/utils.py

@@ -313,19 +313,15 @@ def get_safety_ignore_dict(image_uri, framework, python_version, job_type):
             else (
                 "inference-neuronx"
                 if "neuronx" in image_uri
-                else "inference-neuron"
-                if "neuron" in image_uri
-                else "inference"
+                else "inference-neuron" if "neuron" in image_uri else "inference"
             )
         )
 
     if job_type == "training":
         job_type = (
             "training-neuronx"
             if "neuronx" in image_uri
-            else "training-neuron"
-            if "neuron" in image_uri
-            else "training"
+            else "training-neuron" if "neuron" in image_uri else "training"
         )
 
     if "habana" in image_uri:

test/dlc_tests/sanity/quick_checks/test_required_fixture.py

@@ -136,8 +136,8 @@ def _update_test_fixtures_mapping(file_to_check, test_fixtures_mapping):
             if re.match(test_func_pattern, line):
                 function_name = re.match(test_func_pattern, line).group(1)
                 # Map list of fixtures per tests
-                test_fixtures_mapping[
-                    f"{os.path.basename(file_to_check)}::{function_name}"
-                ] = fixture_list
+                test_fixtures_mapping[f"{os.path.basename(file_to_check)}::{function_name}"] = (
+                    fixture_list
+                )
                 # Empty test_fixtures list for the next test method
                 fixture_list = []

test/dlc_tests/sanity/test_dlc_labels.py

@@ -50,9 +50,7 @@ def test_dlc_standard_labels(image, region):
     arch_type = (
         "graviton"
         if test_utils.is_graviton_architecture()
-        else "arm64"
-        if test_utils.is_arm64_architecture()
-        else "x86"
+        else "arm64" if test_utils.is_arm64_architecture() else "x86"
     )
 
     contributor = test_utils.get_contributor_from_image_uri(image)

test/dlc_tests/sanity/test_pre_release.py

@@ -1204,9 +1204,9 @@ def test_core_package_version(image):
         package_name = package_name.lower()
         installed_version = None
         if package_name not in installed_package_version_dict:
-            violation_data[
-                package_name
-            ] = f"Package: {package_name} not installed in {installed_package_version_dict}"
+            violation_data[package_name] = (
+                f"Package: {package_name} not installed in {installed_package_version_dict}"
+            )
         else:
             installed_version = Version(installed_package_version_dict[package_name])
         if installed_version and installed_version not in SpecifierSet(
@@ -1279,15 +1279,15 @@ def test_package_version_regression_in_image(image):
     violating_packages = {}
     for package_name, version_in_released_image in released_image_package_version_dict.items():
         if package_name not in current_image_package_version_dict:
-            violating_packages[
-                package_name
-            ] = "Not present in the image that is being currently built."
+            violating_packages[package_name] = (
+                "Not present in the image that is being currently built."
+            )
             continue
         version_in_current_image = current_image_package_version_dict[package_name]
         if Version(version_in_released_image) > Version(version_in_current_image):
-            violating_packages[
-                package_name
-            ] = f"Version in already released image: {version_in_released_image} is greater that version in current image: {version_in_current_image}"
+            violating_packages[package_name] = (
+                f"Version in already released image: {version_in_released_image} is greater that version in current image: {version_in_current_image}"
+            )
 
     assert (
         not violating_packages

test/dlc_tests/sanity/test_safety_check.py

@@ -1027,9 +1027,7 @@ def _get_safety_ignore_list(image_uri):
                     else (
                         "inference-neuronx"
                         if "inference-neuronx" in image_uri
-                        else "inference-neuron"
-                        if "inference-neuron" in image_uri
-                        else "inference"
+                        else "inference-neuron" if "inference-neuron" in image_uri else "inference"
                     )
                 )
             )