Make pmg and ASE optional dependencies.

materialsvirtuallab · Jun 20, 2023 · 25766ec · 25766ec
1 parent 938ab4a
commit 25766ec
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/requirements-ci.txt b/requirements-ci.txt
@@ -7,3 +7,5 @@ types-tabulate
 pre-commit
 ruff
 black
+pymatgen==2023.5.31
+ase==3.22.1
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,3 @@
 dgl==1.1.0
 torch==2.0.1
-pymatgen==2023.5.31
-ase==3.22.1
 pytorch_lightning==2.0.3
diff --git a/setup.py b/setup.py
@@ -43,6 +43,10 @@
         "pytorch_lightning",
         "torch",
     ),
+    extra_requires={
+        "ase": ["ase>=3.22.1"],
+        "pymatgen": ["pymatgen>=2023.5.31"],
+    },
     classifiers=[
         "Development Status :: 4 - Beta",
         "Intended Audience :: Science/Research",