[fix] code format

TUDB-Labs · Jul 30, 2024 · 88ff7cd · 88ff7cd
1 parent ab6b3b3
commit 88ff7cd
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/test_moe_layer.py b/tests/test_moe_layer.py
@@ -73,7 +73,7 @@ def test_llama_forward(self):
                 input = torch.zeros(shape)
                 output: torch.Tensor = moe_layer(input)
                 self.assertEqual(output.shape, shape)
-    
+
     def test_phi_forward(self):
         mlp_layer = PhiMLP(
             PhiConfig(
@@ -84,9 +84,7 @@ def test_phi_forward(self):
                 num_attention_heads=2,
             )
         )
-        moe_layer = dummy_moe_layer(
-            "phi", mlp_layer, hidden_size, ["fc1", "fc2"]
-        )
+        moe_layer = dummy_moe_layer("phi", mlp_layer, hidden_size, ["fc1", "fc2"])
         for shape in dummy_test_shapes(hidden_size):
             with self.subTest(f"test for shape = {shape}"):
                 input = torch.zeros(shape)