Changed edge indices to LongTensor

GreeleyGroup · Sep 15, 2023 · 32466b7 · 32466b7
1 parent 9bd23b1
commit 32466b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/featurizers.py b/src/featurizers.py
@@ -505,7 +505,7 @@ def featurize_graph(self, atoms_graph):
         self._feat_tensor = self.encoder.transform(bond_dist_arr)
 
         # Create list of edge indices
-        self._edge_indices = torch.Tensor(list(atoms_graph.graph.edges())).view(2, -1)
+        self._edge_indices = torch.LongTensor(list(atoms_graph.graph.edges())).view(2, -1)
 
     @property
     def feat_tensor(self):