Update batch_gen.py

rel_coords: small bug fix of the reference nodes
BenjaminFiltjens · May 17, 2022 · aca5841 · aca5841
1 parent 761169b
commit aca5841
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/batch_gen.py b/batch_gen.py
@@ -7,7 +7,7 @@
 
 def get_features(sample):
     disps = get_displacements(sample)
-    rel_coords = get_relative_coordinates(sample)
+    rel_coords = get_relative_coordinates(sample, references=(0))
     sample = np.concatenate([disps, rel_coords], axis=0)
     return sample