fix: format with black

Author: Shibo Xing

tests/conftest.py

@@ -88,6 +88,7 @@
 
 PYTORCH_RENEWED_GPU = "ml.g4dn.xlarge"
 
+
 def pytest_addoption(parser):
     parser.addoption("--sagemaker-client-config", action="store", default=None)
     parser.addoption("--sagemaker-runtime-config", action="store", default=None)
@@ -257,9 +258,13 @@ def huggingface_pytorch_training_py_version(huggingface_pytorch_training_version
 
 
 @pytest.fixture(scope="module")
-def huggingface_training_compiler_pytorch_version(huggingface_training_compiler_version):
+def huggingface_training_compiler_pytorch_version(
+    huggingface_training_compiler_version,
+):
     versions = _huggingface_base_fm_version(
-        huggingface_training_compiler_version, "pytorch", "huggingface_training_compiler"
+        huggingface_training_compiler_version,
+        "pytorch",
+        "huggingface_training_compiler",
     )
     if not versions:
         pytest.skip(
@@ -270,9 +275,13 @@ def huggingface_training_compiler_pytorch_version(huggingface_training_compiler_
 
 
 @pytest.fixture(scope="module")
-def huggingface_training_compiler_tensorflow_version(huggingface_training_compiler_version):
+def huggingface_training_compiler_tensorflow_version(
+    huggingface_training_compiler_version,
+):
     versions = _huggingface_base_fm_version(
-        huggingface_training_compiler_version, "tensorflow", "huggingface_training_compiler"
+        huggingface_training_compiler_version,
+        "tensorflow",
+        "huggingface_training_compiler",
     )
     if not versions:
         pytest.skip(
@@ -294,19 +303,25 @@ def huggingface_training_compiler_tensorflow_py_version(
 
 
 @pytest.fixture(scope="module")
-def huggingface_training_compiler_pytorch_py_version(huggingface_training_compiler_pytorch_version):
+def huggingface_training_compiler_pytorch_py_version(
+    huggingface_training_compiler_pytorch_version,
+):
     return "py38"
 
 
 @pytest.fixture(scope="module")
-def huggingface_pytorch_latest_training_py_version(huggingface_training_pytorch_latest_version):
+def huggingface_pytorch_latest_training_py_version(
+    huggingface_training_pytorch_latest_version,
+):
     return (
         "py38" if Version(huggingface_training_pytorch_latest_version) >= Version("1.9") else "py36"
     )
 
 
 @pytest.fixture(scope="module")
-def huggingface_pytorch_latest_inference_py_version(huggingface_inference_pytorch_latest_version):
+def huggingface_pytorch_latest_inference_py_version(
+    huggingface_inference_pytorch_latest_version,
+):
     return (
         "py38"
         if Version(huggingface_inference_pytorch_latest_version) >= Version("1.9")
@@ -482,7 +497,8 @@ def pytorch_ddp_py_version():
 
 
 @pytest.fixture(
-    scope="module", params=["1.10", "1.10.0", "1.10.2", "1.11", "1.11.0", "1.12", "1.12.0"]
+    scope="module",
+    params=["1.10", "1.10.0", "1.10.2", "1.11", "1.11.0", "1.12", "1.12.0"],
 )
 def pytorch_ddp_framework_version(request):
     return request.param
@@ -515,6 +531,7 @@ def gpu_instance_type(sagemaker_session, request):
     else:
         return "ml.p3.2xlarge"
 
+
 @pytest.fixture()
 def gpu_pytorch_instance_type(sagemaker_session, request):
     if "pytorch_inference_version" in request.fixturenames:
@@ -531,6 +548,7 @@ def gpu_pytorch_instance_type(sagemaker_session, request):
     else:
         return "ml.p3.2xlarge"
 
+
 @pytest.fixture(scope="session")
 def gpu_instance_type_list(sagemaker_session, request):
     region = sagemaker_session.boto_session.region_name

tests/unit/sagemaker/image_uris/test_dlc_frameworks.py

@@ -73,7 +73,7 @@ def _test_image_uris(
 
     TYPES_AND_PROCESSORS = INSTANCE_TYPES_AND_PROCESSORS
     if framework == "pytorch" and Version(fw_version) >= Version("1.13"):
-        '''Handle P2 deprecation'''
+        """Handle P2 deprecation"""
         TYPES_AND_PROCESSORS = RENEWED_PYTORCH_INSTANCE_TYPES_AND_PROCESSORS
 
     for instance_type, processor in TYPES_AND_PROCESSORS:

tests/unit/test_pytorch.py

@@ -302,7 +302,12 @@ def test_create_model_with_custom_image(name_from_base, sagemaker_session):
 @patch("sagemaker.estimator.name_from_base", return_value=JOB_NAME)
 @patch("time.time", return_value=TIME)
 def test_pytorch(
-    time, name_from_base, sagemaker_session, pytorch_inference_version, pytorch_inference_py_version, gpu_pytorch_instance_type
+    time,
+    name_from_base,
+    sagemaker_session,
+    pytorch_inference_version,
+    pytorch_inference_py_version,
+    gpu_pytorch_instance_type,
 ):
     pytorch = PyTorch(
         entry_point=SCRIPT_PATH,
@@ -356,7 +361,12 @@ def test_pytorch(
 
 @patch("sagemaker.utils.repack_model", MagicMock())
 @patch("sagemaker.utils.create_tar_file", MagicMock())
-def test_model(sagemaker_session, pytorch_inference_version, pytorch_inference_py_version, gpu_pytorch_instance_type):
+def test_model(
+    sagemaker_session,
+    pytorch_inference_version,
+    pytorch_inference_py_version,
+    gpu_pytorch_instance_type,
+):
     model = PyTorchModel(
         MODEL_DATA,
         role=ROLE,
@@ -429,7 +439,7 @@ def test_model_custom_serialization(
     sagemaker_session,
     pytorch_inference_version,
     pytorch_inference_py_version,
-    gpu_pytorch_instance_type
+    gpu_pytorch_instance_type,
 ):
     model = PyTorchModel(
         MODEL_DATA,