Fix density validator (#859)

materialsproject · Oct 13, 2023 · aad9f2e · aad9f2e
1 parent e05aec3
commit aad9f2e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/emmet-core/emmet/core/tasks.py b/emmet-core/emmet/core/tasks.py
@@ -104,7 +104,11 @@ def set_density_from_structure(cls, values):
         # specified. This might happen when importing an older atomate2-format
         # TaskDocument.
         if not values.get("density", None):
-            values["density"] = values["structure"].density
+            if isinstance(values["structure"], dict):
+                values["density"] = values["structure"].get("density", None)
+            else:
+                values["density"] = values["structure"].density
+
         return values
 
     @classmethod