fix: mapping site.index instead of just site

IntelLabs · Dec 20, 2024 · d4e9e36 · d4e9e36
1 parent 3f208a1
commit d4e9e36
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/matsciml/datasets/utils.py b/matsciml/datasets/utils.py
@@ -881,7 +881,7 @@ def _all_sites_have_neighbors(neighbors):
         for src_idx, dst_sites in enumerate(neighbors):
             for site in dst_sites:
                 all_src.append(src_idx)
-                all_dst.append(site)
+                all_dst.append(site.index)
                 all_images.append(site.image)
     if any([len(obj) == 0 for obj in [all_src, all_dst, all_images]]):
         raise ValueError(