From 913d8faeb8f8eb85340bd79b460c2995a7327108 Mon Sep 17 00:00:00 2001 From: Johnny1882 Date: Sun, 26 Jan 2025 22:47:57 +0000 Subject: [PATCH] black format fixed --- .../module/transforms/optical/test_optical_module.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/test/passes/module/transforms/optical/test_optical_module.py b/test/passes/module/transforms/optical/test_optical_module.py index f52acd559..2041b9ccf 100644 --- 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__':