undo temporary fix

atomistic-machine-learning · Aug 28, 2024 · bd86896 · bd86896
1 parent e5564aa
commit bd86896
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/schnetpack/data/atoms.py b/src/schnetpack/data/atoms.py
@@ -359,9 +359,7 @@ def _get_properties(
             properties[structure.cell] = (
                 torch.tensor(row["cell"][None].copy()) * self.distance_conversion
             )
-            properties[structure.pbc] = torch.tensor(
-                row["pbc"].tolist()
-            )  # todo: check this RuntimeError: Could not infer dtype of numpy.bool
+            properties[structure.pbc] = torch.tensor(row["pbc"])
 
         return properties