Fix tests (#22)

* Used bleeding edge of sinabs for tests Closes #21
neuromorphs · Oct 21, 2023 · c7aba27 · c7aba27
1 parent dc1d6aa
commit c7aba27
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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():

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