fixed NIRNode import issue #85

neuromorphs · Mar 27, 2024 · 2ce271e · 2ce271e
1 parent 8af5be2
commit 2ce271e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/nir/ir/__init__.py b/nir/ir/__init__.py
@@ -81,6 +81,8 @@ def dict2NIRNode(data_dict: Dict[str, Any]) -> NIRNode:
     "IF",
     "LI",
     "LIF",
+    # node
+    "NIRNode",
     # pooling
     "AvgPool2d",
     "SumPool2d",

diff --git a/tests/test_ir.py b/tests/test_ir.py
@@ -9,6 +9,10 @@ def test_has_version():
     assert hasattr(nir, "__version__")
 
 
+def test_has_NIRNode():
+    assert hasattr(nir, "NIRNode")
+
+
 def test_eq():
     a = nir.Input(np.array([2, 3]))
     a2 = nir.Input(np.array([2, 3]))