Make style

huggingface · githubnemo · Aug 27, 2025 · Apr 24, 2025 · May 11, 2025 · May 16, 2025
commit 3fca72a4425e6ed0306a802f4de40e47c8a14c08
diff --git a/tests/test_custom_models.py b/tests/test_custom_models.py
@@ -2477,9 +2477,9 @@ def test_multirank_2(self):
                 if isinstance(module, BaseTunerLayer):
                     rank_expected = rank_pattern.get(key, r)
                     rank_current = module.lora_A[adapter].weight.shape[0]
-                    assert (
-                        rank_current == rank_expected
-                    ), f"Rank {rank_current} is not equal to expected {rank_expected}"
+                    assert rank_current == rank_expected, (
+                        f"Rank {rank_current} is not equal to expected {rank_expected}"
+                    )
 
 
 class TestRepr(unittest.TestCase):