chore(quality): style here we go again ... again

Author: mfuntowicz

tests/integrations/test_hub_kernels.py

@@ -7,16 +7,16 @@
 
 @require_kernels
 class HubKernelsTests(unittest.TestCase):
-
     def test_disable_hub_kernels(self):
         """
         Test that _kernels_enabled is False when USE_HUB_KERNELS when USE_HUB_KERNELS=OFF
         """
-        with patch.dict(os.environ, {'USE_HUB_KERNELS': 'ON'}):
+        with patch.dict(os.environ, {"USE_HUB_KERNELS": "ON"}):
             # Re-import to ensure the environment variable takes effect
             import importlib
 
             from transformers.integrations import hub_kernels
+
             importlib.reload(hub_kernels)
 
             # Verify that kernels are disabled
@@ -27,12 +27,13 @@ def test_enable_hub_kernels_default(self):
         Test that _kernels_enabled is True when USE_HUB_KERNELS is not provided (default behavior)
         """
         # Remove USE_HUB_KERNELS from the environment if it exists
-        env_without_hub_kernels = {k: v for k, v in os.environ.items() if k != 'USE_HUB_KERNELS'}
+        env_without_hub_kernels = {k: v for k, v in os.environ.items() if k != "USE_HUB_KERNELS"}
         with patch.dict(os.environ, env_without_hub_kernels, clear=True):
             # Re-import to ensure the environment variable change takes effect
             import importlib
 
             from transformers.integrations import hub_kernels
+
             importlib.reload(hub_kernels)
 
             # Verify that kernels are enabled by default
@@ -42,27 +43,29 @@ def test_enable_hub_kernels_on(self):
         """
         Test that _kernels_enabled is True when USE_HUB_KERNELS=ON
         """
-        with patch.dict(os.environ, {'USE_HUB_KERNELS': 'ON'}):
+        with patch.dict(os.environ, {"USE_HUB_KERNELS": "ON"}):
             # Re-import to ensure the environment variable takes effect
             import importlib
 
             from transformers.integrations import hub_kernels
+
             importlib.reload(hub_kernels)
 
             # Verify that kernels are enabled
             self.assertTrue(hub_kernels._kernels_enabled)
 
-    @patch('kernels.use_kernel_forward_from_hub')
+    @patch("kernels.use_kernel_forward_from_hub")
     def test_use_kernel_forward_from_hub_not_called_when_disabled(self, mocked_use_kernel_forward):
         """
         Test that kernels.use_kernel_forward_from_hub is not called when USE_HUB_KERNELS is disabled
         """
         # Set environment variable to disable hub kernels
-        with patch.dict(os.environ, {'USE_HUB_KERNELS': 'OFF'}):
+        with patch.dict(os.environ, {"USE_HUB_KERNELS": "OFF"}):
             # Re-import to ensure the environment variable takes effect
             import importlib
 
             from transformers.integrations import hub_kernels
+
             importlib.reload(hub_kernels)
 
             # Call the function with a test layer name
@@ -78,18 +81,19 @@ class FooClass:
             result = decorator(FooClass)
             self.assertIs(result, FooClass)
 
-    @patch('kernels.use_kernel_forward_from_hub')
+    @patch("kernels.use_kernel_forward_from_hub")
     def test_use_kernel_forward_from_hub_called_when_enabled_default(self, mocked_use_kernel_forward):
         """
         Test that kernels.use_kernel_forward_from_hub is called when USE_HUB_KERNELS is not set (default)
         """
         # Remove USE_HUB_KERNELS from the environment if it exists
-        env_without_hub_kernels = {k: v for k, v in os.environ.items() if k != 'USE_HUB_KERNELS'}
+        env_without_hub_kernels = {k: v for k, v in os.environ.items() if k != "USE_HUB_KERNELS"}
         with patch.dict(os.environ, env_without_hub_kernels, clear=True):
             # Re-import to ensure the environment variable change takes effect
             import importlib
 
             from transformers.integrations import hub_kernels
+
             importlib.reload(hub_kernels)
 
             # Call the function with a test layer name
@@ -98,22 +102,21 @@ def test_use_kernel_forward_from_hub_called_when_enabled_default(self, mocked_us
             # Verify that the kernels function was called once with the correct argument
             mocked_use_kernel_forward.assert_called_once_with("FooLayer")
 
-    @patch('kernels.use_kernel_forward_from_hub')
+    @patch("kernels.use_kernel_forward_from_hub")
     def test_use_kernel_forward_from_hub_called_when_enabled_on(self, mocked_use_kernel_forward):
         """
         Test that kernels.use_kernel_forward_from_hub is called when USE_HUB_KERNELS=ON
         """
-        with patch.dict(os.environ, {'USE_HUB_KERNELS': 'ON'}):
+        with patch.dict(os.environ, {"USE_HUB_KERNELS": "ON"}):
             # Re-import to ensure the environment variable change takes effect
             import importlib
 
             from transformers.integrations import hub_kernels
+
             importlib.reload(hub_kernels)
 
             # Call the function with a test layer name
             hub_kernels.use_kernel_forward_from_hub("FooLayer")
 
             # Verify that the kernels function was called once with the correct argument
             mocked_use_kernel_forward.assert_called_once_with("FooLayer")
-
-