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

[Test] remove/don't generate tmp files in main folder

parent e653b671
No related branches found
No related tags found
No related merge requests found
......@@ -57,9 +57,9 @@ def test_absorb_transp_into_flatten(perm, shape, ishape, data_layout):
model = model.transform(InferDataLayouts())
model = model.transform(GiveUniqueNodeNames())
model = model.transform(GiveReadableTensorNames())
model.save("test.onnx")
# model.save("test.onnx")
model_transformed = model.transform(AbsorbTransposeIntoFlatten())
model_transformed.save("test2.onnx")
# model_transformed.save("test2.onnx")
# verify transformation
inp_values = np.random.uniform(low=-1, high=1, size=tuple(ishape)).astype(
......
# import os
import os
import onnx
from finn.util.test import get_test_model_trained
import brevitas.onnx as bo
......@@ -57,4 +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)
os.remove(export_onnx_path)
......@@ -61,4 +61,4 @@ def test_hls_random_mlp_maker(bitwidth):
ret = create.hls_random_mlp_maker(layer_spec)
assert len(ret.graph.node) == 5
ret.save("mlp-%s.onnx" % str(bitwidth))
# ret.save("mlp-%s.onnx" % str(bitwidth))
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