From 7d6c934f72a3a4304b4910f423a343f425925da3 Mon Sep 17 00:00:00 2001 From: Nick Fraser Date: Mon, 27 Jan 2025 08:55:19 +0000 Subject: [PATCH] Fix: precommit --- tests/brevitas/core/test_trunc_int_quant.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/brevitas/core/test_trunc_int_quant.py b/tests/brevitas/core/test_trunc_int_quant.py index ea386c0ae..a064b0676 100644 --- a/tests/brevitas/core/test_trunc_int_quant.py +++ b/tests/brevitas/core/test_trunc_int_quant.py @@ -11,12 +11,12 @@ from brevitas.core.function_wrapper import RoundSte from brevitas.core.function_wrapper import TensorClamp from brevitas.core.quant import TruncIntQuant +from brevitas.core.restrict_val import PowerOfTwoRestrictValue +from brevitas.core.scaling import RuntimeStatsScaling from brevitas.core.scaling import TruncMsbScaling from brevitas.core.scaling import TruncPowerOfTwoIntScaling from brevitas.core.scaling import TruncScalingWrapper -from brevitas.core.scaling import RuntimeStatsScaling from brevitas.core.stats import AbsMax -from brevitas.core.restrict_val import PowerOfTwoRestrictValue from tests.brevitas.core.bit_width_fixture import * # noqa from tests.brevitas.core.int_quant_fixture import * # noqa