Skip to content
Snippets Groups Projects
Commit 76a4d609 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Test] remove generated .onnx files

parent 7695eb07
No related branches found
No related tags found
No related merge requests found
import os
import numpy as np
import torch
import brevitas.onnx as bo
......@@ -41,3 +42,4 @@ def test_brevitas_act_export(abits, narrow_range, max_val):
inp_tensor = torch.from_numpy(inp_tensor).float()
expected = b_act.forward(inp_tensor).detach().numpy()
assert np.isclose(produced, expected, atol=1e-3).all()
os.remove(export_onnx_path)
import os
import onnx
from finn.util.test import get_test_model_trained
import brevitas.onnx as bo
......@@ -56,3 +57,4 @@ def test_topk_insert(k):
output_pysim_topk = output_pysim_topk.astype(np.int).flatten()
assert np.array_equal(output_golden_topk, output_pysim_topk)
os.remove(export_onnx_path)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment