Fixing imports

Xilinx · Jan 19, 2024 · dafedb9 · dafedb9
1 parent 85871f0
commit dafedb9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/brevitas_examples/test_examples_import.py b/tests/brevitas_examples/test_examples_import.py
@@ -4,7 +4,7 @@
 import pytest
 
 from brevitas.quant.scaled_int import Int8AccumulatorAwareWeightQuant
-from brevitas.quant.scaled_int import Int8ImprovedAccumulatorAwareWeightQuant
+from brevitas.quant.scaled_int import Int8AccumulatorAwareZeroCenterWeightQuant
 from brevitas.quant.scaled_int import Int8WeightPerChannelFloat
 
 
@@ -46,7 +46,7 @@ def test_import_stt():
     [
         Int8WeightPerChannelFloat,
         Int8AccumulatorAwareWeightQuant,
-        Int8ImprovedAccumulatorAwareWeightQuant])
+        Int8AccumulatorAwareZeroCenterWeightQuant])
 def test_super_resolution_float_and_quant_models_match(upscale_factor, num_channels, weight_quant):
     import brevitas.config as config
     from brevitas_examples.super_resolution.models import float_espcn