Fix formatting

pulp-platform · Feb 13, 2024 · 6b0c166 · 6b0c166
1 parent dfe0c4f
commit 6b0c166
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/test/NnxTestClasses.py b/test/NnxTestClasses.py
@@ -254,7 +254,10 @@ def from_conf(
                 ).type(torch.int32)
             if global_shift is None:
                 global_shift = torch.Tensor([0]).type(torch.int32)
-                conv_kwargs = {**conf.__dict__, "out_type": NeuralEngineFunctionalModel.ACCUMULATOR_TYPE}
+                conv_kwargs = {
+                    **conf.__dict__,
+                    "out_type": NeuralEngineFunctionalModel.ACCUMULATOR_TYPE,
+                }
                 output = NeuralEngineFunctionalModel().convolution(
                     input,
                     weight,
@@ -264,7 +267,9 @@ def from_conf(
                     verbose=False,
                     **conv_kwargs,
                 )
-                global_shift = NnxTestGenerator._calculate_global_shift(output, conf.out_type)
+                global_shift = NnxTestGenerator._calculate_global_shift(
+                    output, conf.out_type
+                )
 
         output = NeuralEngineFunctionalModel().convolution(
             input, weight, scale, bias, global_shift, verbose=verbose, **conf.__dict__