Move import to top

Xilinx · Jan 13, 2025 · beb2cbc · beb2cbc
1 parent c599f31
commit beb2cbc
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/brevitas/graph/equalize.py b/src/brevitas/graph/equalize.py
@@ -39,6 +39,7 @@
 from brevitas.nn.equalized_layer import INPUT_NAMES
 from brevitas.nn.equalized_layer import RotatedModule
 from brevitas.nn.quant_scale_bias import ScaleBias
+from brevitas.utils.python_utils import recurse_getattr
 from brevitas.utils.rotation_utils import RotationWeightParametrization
 from brevitas.utils.torch_utils import KwargsForwardHook
 
@@ -1447,9 +1448,6 @@ def _apply_rotate(
     return rewriters
 
 
-from brevitas.utils.python_utils import recurse_getattr
-
-
 def _untie_parameters_with_parametrizations(model: torch.nn.Module):
     # get ALL model parameters and their names
     all_named_parameters = {