black format fixed

DeepWok · Jan 26, 2025 · 913d8fa · 913d8fa
1 parent bd44a07
commit 913d8fa
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/test/passes/module/transforms/optical/test_optical_module.py b/test/passes/module/transforms/optical/test_optical_module.py
@@ -26,6 +26,7 @@
 
 from train_mnist_cnn import test, train, Net
 
+
 class Net(nn.Module):
     def __init__(self):
         super(Net, self).__init__()
@@ -89,15 +90,8 @@ def test_optical_module_transform_pass():
     optical_module_transform_pass(model, pass_args)
     # torch.save(onn_cnn, "mase_output/onn_cnn.pt")
 
-test_optical_module_transform_pass()
-
-
-
-
-
-
-
 
+test_optical_module_transform_pass()
 
 
 # if __name__ == '__main__':