From 32466b7de6f2b2fbf7e6ada5d5c7b821cdfae1a0 Mon Sep 17 00:00:00 2001 From: Gaurav S Deshmukh Date: Fri, 15 Sep 2023 00:34:22 -0400 Subject: [PATCH] Changed edge indices to LongTensor --- src/featurizers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/featurizers.py b/src/featurizers.py index 9a37b73..9597295 100644 --- 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):