scaffold

gotennet_pytorch/__init__.py

@@ -0,0 +1 @@
+from gotennet_pytorch import gotennet

gotennet_pytorch/gotennet.py

@@ -0,0 +1,20 @@
+import torch
+import torch.nn.functional as F
+from torch.nn import Module, ModuleList
+
+# helper functions
+
+def exists(v):
+    return v is not None
+
+def default(v, d):
+    return v if exists(v) else d
+
+# main class
+
+class GotenNet(Module):
+    def __init__(self):
+        super().__init__()
+
+    def forward(self, x):
+        return x

pyproject.toml

@@ -50,6 +50,9 @@ dev-dependencies = [
     "pytest-cov>=5.0.0",
 ]
 
+[tool.pytest.ini_options]
+pythonpath = ["."]
+
 [tool.hatch.metadata]
 allow-direct-references = true
 

tests/test_gotennet.py

@@ -1,3 +1,6 @@
 
+from gotennet_pytorch.gotennet import GotenNet
+
 def test_equivariant():
+    model = GotenNet()
     assert True