From d0f22ee57720a2017bf25c20b8d7498be81b9a8a Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Thu, 14 May 2020 11:09:51 +0100
Subject: [PATCH] [Test] remove gen onnx file from
test_brevitas_relu_act_export
---
tests/brevitas/test_brevitas_relu_act_export.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/tests/brevitas/test_brevitas_relu_act_export.py b/tests/brevitas/test_brevitas_relu_act_export.py
index 8a71ea54c..4c450243b 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)
--
GitLab