chore: 🚨 Fix lint errors

Author: sodre

src/sagemaker/local/image.py

@@ -50,7 +50,11 @@
 S3_ENDPOINT_URL_ENV_NAME = "S3_ENDPOINT_URL"
 
 # SELinux Enabled
-SELINUX_ENABLED = os.environ.get("SAGEMAKER_LOCAL_SELINUX_ENABLED", "False").lower() in ["1", "true", "yes"]
+SELINUX_ENABLED = os.environ.get("SAGEMAKER_LOCAL_SELINUX_ENABLED", "False").lower() in [
+    "1",
+    "true",
+    "yes",
+]
 
 logger = logging.getLogger(__name__)
 

tests/unit/sagemaker/local/test_local_image.py

@@ -571,19 +571,15 @@ def test_train_local_intermediate_output(get_data_source_instance, tmpdir, sagem
 
 @patch("platform.system", Mock(return_value="Linux"))
 @patch("sagemaker.local.image.SELINUX_ENABLED", Mock(return_value=True))
-def test_container_selinux_has_label(tmpdir, sagemaker_session):
-    instance_count = 1
-    image = "my-image"
+def test_container_selinux_has_label(tmpdir):
     volume = _Volume(str(tmpdir), "/opt/ml/model")
 
     assert volume.map.endswith(":z")
 
 
 @patch("platform.system", Mock(return_value="Darwin"))
 @patch("sagemaker.local.image.SELINUX_ENABLED", Mock(return_value=True))
-def test_container_has_selinux_no_label(tmpdir, sagemaker_session, monkeypatch):
-    instance_count = 1
-    image = "my-image"
+def test_container_has_selinux_no_label(tmpdir):
     volume = _Volume(str(tmpdir), "/opt/ml/model")
 
     assert not volume.map.endswith(":z")