diff --git a/prereq.txt b/prereq.txt index 75d0c81e..2d6fae3f 100644 --- a/prereq.txt +++ b/prereq.txt @@ -1,4 +1,4 @@ numpy>=1.20 -torch>=2.1, <2.3 # Max due to tsai -tsai +torch>=2.1 # Min due to opacus, tsai 0.4.1+ supports latest PyTorch +tsai>=0.4.1 wheel>=0.40 diff --git a/setup.cfg b/setup.cfg index e4623326..e69d8171 100644 --- a/setup.cfg +++ b/setup.cfg @@ -34,12 +34,12 @@ python_requires = >=3.9 install_requires = importlib-metadata pandas>=2.1 # min due to lifelines - torch>=2.1, <2.3 # Max due to tsai, min due to opacus + torch>=2.3 # Min due to opacus, tsai 0.4.1+ supports latest PyTorch scikit-learn>=1.2 nflows>=0.14 - numpy>=1.20, <2.0 + numpy>=1.20, <2.0 # max due to lifelines 0.29.0 requirement lifelines>=0.29.0, <0.30.0 # max due to xgbse - opacus>=1.3, <1.5.4 # 1.5.4 introduces RMSNorm error + opacus>=1.5.4 # RMSNorm support required for PyTorch 2.x compatibility networkx>2.0,<3.0 decaf-synthetic-data>=0.1.7 optuna>=3.1 @@ -55,17 +55,17 @@ install_requires = pgmpy<1.0 redis pycox - xgbse>=0.3.1 + xgbse==0.3.3 pykeops fflows monai - tsai; python_version>"3.7" + tsai>=0.4.1; python_version>"3.7" be-great>=0.0.5;python_version>="3.9" arfpy - fastcore<1.8 # Required by breaking change in fastai 2.8 which is needed by tsai - fastai<2.8 # Required by breaking change in fastai 2.8 which is needed by tsai - transformers<4.33.0 # Required by great plugin while tranformers >4.33.0 is not compatible with torch <2.5 - accelerate<0.20.4 # Required for transformers <4.33.0 + fastcore>=1.8 # Required for tsai 0.4.1+ compatibility + fastai>=2.8 # Required for tsai 0.4.1+ compatibility + transformers # Now compatible with latest PyTorch + accelerate # Now compatible with latest PyTorch and transformers [options.packages.find]