diff --git a/pyproject.toml b/pyproject.toml
index f113677..5e1517e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,9 +29,12 @@ dependencies = [
     "ase>=3.23.0",
     "azure-identity",
     "azure-storage-blob",
+    "scikit-learn",
     "deprecated",
     "e3nn>=0.5.0",
-    "emmet-core>=0.84",
+    "atomate2",
+    "emmet-core>=0.84; python_version >= '3.10'",
+    "emmet-core<0.84; python_version < '3.10'",
     "loguru",
     "mp-api",
     "numpy<2",