refactor: making mp dataset read cell directly

IntelLabs · Nov 25, 2024 · 039a84d · 039a84d
1 parent dbcd044
commit 039a84d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/matsciml/datasets/materials_project/dataset.py b/matsciml/datasets/materials_project/dataset.py
@@ -146,6 +146,7 @@ def _parse_structure(
             "lattice_params": lattice_params,
         }
         return_dict["lattice_features"] = lattice_features
+        return_dict["cell"] = structure.lattice.matrix
 
     def _parse_symmetry(
         self,