Merged sinabs test changes

neuromorphs · Oct 21, 2023 · 5845167 · 5845167
2 parents ec8cded + c7aba27
commit 5845167
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/tests/test_from_nir.py b/tests/test_from_nir.py
@@ -56,12 +56,9 @@ def test_extract_empty():
 
 
 def test_extract_illegal_name():
-    graph = nir.NIRGraph(
-        {"i": nir.Input(np.ones((1, 1))), "a.b": nir.Linear(np.ones((1, 1)))},
-        [("i", "a.b")],
-    )
+    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():

diff --git a/tests/test_requirements.txt b/tests/test_requirements.txt
@@ -3,4 +3,4 @@ torch
 nir
 ruff
 git+https://github.com/norse/norse
-sinabs
+git+https://github.com/synsense/sinabs@dev/nir
-Original file line number
+Diff line change
@@ Expand Up / @@ -3,4 +3,4 @@ torch @@
     nir
     ruff
     git+https://github.com/norse/norse
-    sinabs
+    git+https://github.com/synsense/sinabs@dev/nir