Convert listconfig to list before adding to get total include_key

mir-group · Nov 18, 2024 · f131ae1 · f131ae1
1 parent da321ec
commit f131ae1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nequip/data/ase.py b/nequip/data/ase.py
@@ -47,7 +47,7 @@ def from_ase(
         ]  # arguments for from_dict method
     )
     include_keys = list(
-        set(include_keys + ase_all_properties + list(key_mapping.keys()))
+        set(list(include_keys) + ase_all_properties + list(key_mapping.keys()))
         - default_args
         - set(exclude_keys)
     )