Fix lint

pytorch · Oct 29, 2022 · 0bae9de · 0bae9de
1 parent ea0dd3a
commit 0bae9de
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/quantization/fx/test_quantize_fx.py b/test/quantization/fx/test_quantize_fx.py
@@ -5312,9 +5312,9 @@ def forward(self, x):
         x = torch.randn(4, 4, 4, 4)
         # torch.channel_shuffle is equivalent to torch.nn.functional.channel_shuffle
         model_node_pairs = [
-          (M1().eval(), ns.call_module(torch.nn.ChannelShuffle)),
-          (M2().eval(), ns.call_function(torch.channel_shuffle)),
-          (M3().eval(), ns.call_function(torch.channel_shuffle))
+            (M1().eval(), ns.call_module(torch.nn.ChannelShuffle)),
+            (M2().eval(), ns.call_function(torch.channel_shuffle)),
+            (M3().eval(), ns.call_function(torch.channel_shuffle))
         ]
         for m, node in model_node_pairs:
             m = prepare_fx(m, {"": default_qconfig}, example_inputs=(x,))