From 6b0c166d08d2bb82ee0ec357cd136cc6181be893 Mon Sep 17 00:00:00 2001 From: Luka Macan Date: Tue, 13 Feb 2024 14:52:10 +0100 Subject: [PATCH] Fix formatting --- test/NnxTestClasses.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/test/NnxTestClasses.py b/test/NnxTestClasses.py index d94c0fa..edf227b 100644 --- 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__