Merge branch 'materialsvirtuallab:main' into main

materialsvirtuallab · Jun 23, 2024 · 6aaad51 · 6aaad51
2 parents 16abc38 + cfd4874
commit 6aaad51
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/models/test_tensornet.py b/tests/models/test_tensornet.py
@@ -21,6 +21,8 @@ def test_model(self, graph_MoS):
         os.remove("model.pt")
         os.remove("model.json")
         os.remove("state.pt")
+        model = TensorNet(is_intensive=False, equivariance_invariance_group="SO(3)")
+        assert torch.numel(output) == 1
 
     def test_exceptions(self):
         with pytest.raises(ValueError, match="Invalid activation type"):