use torch.linalg.cross to silence UserWarning about missing dim kwarg

#132 (comment)
CederGroupHub · Feb 29, 2024 · fc77430 · fc77430
1 parent 9f6dd8c
commit fc77430
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/chgnet/model/model.py b/chgnet/model/model.py
@@ -807,7 +807,9 @@ def from_graphs(
             else:
                 strain = None
                 lattice = graph.lattice
-            volumes.append(torch.dot(lattice[0], torch.cross(lattice[1], lattice[2])))
+            volumes.append(
+                torch.dot(lattice[0], torch.linalg.cross(lattice[1], lattice[2]))
+            )
             strains.append(strain)
 
             # Bonds