From d8e2587d1f815358329bca722f2ca6708731cd46 Mon Sep 17 00:00:00 2001 From: "Jens E. Pedersen" Date: Sat, 21 Oct 2023 21:24:58 +0200 Subject: [PATCH 1/4] Used bleeding edge of sinabs for tests --- tests/test_requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_requirements.txt b/tests/test_requirements.txt index 351614b..0d00b5d 100644 --- a/tests/test_requirements.txt +++ b/tests/test_requirements.txt @@ -3,4 +3,4 @@ torch nir ruff git+https://github.com/norse/norse -sinabs \ No newline at end of file +git+https://gitlab.com/synsense/sinabs From 51aee39e4553d3fe04b2db794934280a46764922 Mon Sep 17 00:00:00 2001 From: "Jens E. Pedersen" Date: Sat, 21 Oct 2023 21:26:11 +0200 Subject: [PATCH 2/4] Used bleeding edge of sinabs --- tests/test_requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_requirements.txt b/tests/test_requirements.txt index 0d00b5d..6d9a167 100644 --- a/tests/test_requirements.txt +++ b/tests/test_requirements.txt @@ -3,4 +3,4 @@ torch nir ruff git+https://github.com/norse/norse -git+https://gitlab.com/synsense/sinabs +git+https://github.com/synsense/sinabs@dev/nix From 906c803d9288f7466303597d79f23da84522b9b9 Mon Sep 17 00:00:00 2001 From: "Jens E. Pedersen" Date: Sat, 21 Oct 2023 21:32:59 +0200 Subject: [PATCH 3/4] Used bleeding edge of sinabs --- tests/test_requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_requirements.txt b/tests/test_requirements.txt index 6d9a167..50339f8 100644 --- a/tests/test_requirements.txt +++ b/tests/test_requirements.txt @@ -3,4 +3,4 @@ torch nir ruff git+https://github.com/norse/norse -git+https://github.com/synsense/sinabs@dev/nix +git+https://github.com/synsense/sinabs@dev/nir From 9366ce2bf29003bac34c2e6b777b9ec2f678b520 Mon Sep 17 00:00:00 2001 From: "Jens E. Pedersen" Date: Sat, 21 Oct 2023 22:02:16 +0200 Subject: [PATCH 4/4] Fixed test --- tests/test_from_nir.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_from_nir.py b/tests/test_from_nir.py index d6d09ae..3c07d12 100644 --- a/tests/test_from_nir.py +++ b/tests/test_from_nir.py @@ -29,9 +29,9 @@ def test_extract_empty(): def test_extract_illegal_name(): - graph = nir.NIRGraph({"a.b": nir.Input(np.ones(1))}, []) + graph = nir.NIRGraph({"a.b": nir.Input(np.ones(1)), "a.c": nir.Linear(np.array([[1.]]))}, [("a.b", "a.c")]) torch_graph = load(graph, _torch_model_map) - assert "a_b" in torch_graph._modules + assert "a_c" in torch_graph._modules def test_extract_lin():