From 039a84dfedcddaa9ea1cc5e116b5a016ba57845b Mon Sep 17 00:00:00 2001
From: "Lee, Kin Long Kelvin" <kin.long.kelvin.lee@intel.com>
Date: Sun, 24 Nov 2024 16:43:50 -0800
Subject: [PATCH] refactor: making mp dataset read cell directly

---
 matsciml/datasets/materials_project/dataset.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/matsciml/datasets/materials_project/dataset.py b/matsciml/datasets/materials_project/dataset.py
index 763a4308..9913ae7a 100644
--- 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,