Exclude dimension attributes from properties

During creation of the DLite structure.
SINTEF · Oct 23, 2023 · 4fbc7ba · 4fbc7ba
1 parent 93245a8
commit 4fbc7ba
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/oteapi_optimade/dlite/parse.py b/oteapi_optimade/dlite/parse.py
@@ -241,12 +241,16 @@ def get(
 
             # Attributes
             new_structure_attributes.update(
-                structure.attributes.dict(exclude={"species", "assemblies"})
+                structure.attributes.dict(
+                    exclude={"species", "assemblies", "nelements", "nsites"}
+                )
             )
             for key in list(new_structure_attributes):
                 if key.startswith("_"):
                     new_structure_attributes.pop(key)
 
+            # Structure features values are Enum values, so we need to convert them to
+            # their string (true) values
             new_structure_attributes["structure_features"] = [
                 _.value for _ in new_structure_attributes["structure_features"]
             ]