diff --git a/tests/brevitas/test_brevitas_relu_act_export.py b/tests/brevitas/test_brevitas_relu_act_export.py index 8a71ea54c0ed33843c3e0c10e124a19a27cb8ce0..4c450243b9828d1f03f28044c32b6325c678af4c 100644 --- a/tests/brevitas/test_brevitas_relu_act_export.py +++ b/tests/brevitas/test_brevitas_relu_act_export.py @@ -1,3 +1,4 @@ +import os import onnx # noqa import numpy as np import torch @@ -65,3 +66,4 @@ scaling_impl.learned_value": torch.tensor( print("expec:", ", ".join(["{:8.4f}".format(x) for x in expected[0]])) assert np.isclose(produced, expected, atol=1e-3).all() + os.remove(export_onnx_path)