From c9f28d7557ea13d6402fee8473cb53263c3fd762 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Tue, 6 Feb 2024 19:18:52 +0000 Subject: [PATCH] Misc: pre-commit fix --- src/brevitas/graph/equalize.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/brevitas/graph/equalize.py b/src/brevitas/graph/equalize.py index d81335df8..6e4bce6e8 100644 --- a/src/brevitas/graph/equalize.py +++ b/src/brevitas/graph/equalize.py @@ -514,7 +514,8 @@ def _no_equalize(): shape_0 = list_of_act_val_shapes[0] if any(shape_0 != shape for shape in list_of_act_val_shapes): return _no_equalize() - list_of_act_val = list_of_act_val = [transpose(act_val, act_axis) for act_val in list_of_act_val] + list_of_act_val = list_of_act_val = [ + transpose(act_val, act_axis) for act_val in list_of_act_val] srcs_range_act = scale_fn( torch.cat([act_val.reshape(act_val.size(0), -1) for act_val in list_of_act_val], 1)) srcs_range_act = srcs_range_act.to(device=device)