diff --git a/pyproject.toml b/pyproject.toml index f923939..d91e1fb 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -24,12 +24,11 @@ classifiers = [ ] dependencies = [ "numpy<=2.0", - "torch<=2.3.0", - "numba", - "scipy<1.11.1", - "sympy", - "torch<=2.0.1", - "scikit-learn", + "torch<=2.4.0", + "numba<=0.60.0", + "scipy<=1.14.1", + "sympy<=1.13.1", + "scikit-learn<=1.5.1", "psutil", "h5py", "dask", diff --git a/requirements.txt b/requirements.txt index a14158b..dd2668e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,9 +1,9 @@ numpy<=2.0 -numba -scipy -sympy +numba<=0.60.0 +scipy<=1.14.0 +sympy<=1.13.1 torch<=2.4.0 -scikit-learn +scikit-learn<=1.5.1 psutil h5py dask