revert more np related changes

CederGroupHub · Sep 12, 2024 · 8875487 · 8875487
1 parent c9e8e5f
commit 8875487
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -11,7 +11,7 @@ dependencies = [
     "cython>=3",
     "numpy>=1.26,<2",
     "nvidia-ml-py3>=7.352.0",
-    "pymatgen>=2024.9.10",
+    "pymatgen<2024.9.10",
     "torch>=1.11.0",
     "typing-extensions>=4.12",
 ]
@@ -52,7 +52,6 @@ requires = [
     "Cython",
     "setuptools>=65",
     "wheel",
-    "numpy>=2.0.0",
 ]
 build-backend = "setuptools.build_meta"
 

diff --git a/setup.py b/setup.py
@@ -1,12 +1,10 @@
 from __future__ import annotations
 
-import numpy as np
 from setuptools import Extension, setup
 
 ext_modules = [Extension("chgnet.graph.cygraph", ["chgnet/graph/cygraph.pyx"])]
 
 setup(
     ext_modules=ext_modules,
     setup_requires=["Cython"],
-    include_dirs=[np.get_include()],
 )