faulty base branch update removed

Signed-off-by: lstasytis <[email protected]>
Xilinx · Oct 15, 2024 · 0efe48b · 0efe48b
1 parent 4df80fa
commit 0efe48b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/test_brevitas_debug.onnx b/test_brevitas_debug.onnx
diff --git a/tests/brevitas/test_brevitas_debug.py b/tests/brevitas/test_brevitas_debug.py
@@ -35,7 +35,6 @@
 import os
 import torch
 from brevitas.export import export_qonnx
-from brevitas.quant_tensor import _unpack_quant_tensor
 from pkgutil import get_data
 from qonnx.core.modelwrapper import ModelWrapper
 from qonnx.util.cleanup import cleanup as qonnx_cleanup
@@ -91,7 +90,7 @@ def test_brevitas_debug(QONNX_FINN_conversion):
     else:
         assert len(names_common) == 8
     for dbg_name in names_common:
-        tensor_pytorch = _unpack_quant_tensor(dbg_hook.values[dbg_name]).detach().numpy()
+        tensor_pytorch = dbg_hook.values[dbg_name].value.detach().numpy()
         tensor_finn = output_dict[dbg_name]
         assert np.isclose(tensor_finn, tensor_pytorch, atol=1e-5).all()
     os.remove(finn_onnx)