diff --git a/requirements.txt b/requirements.txt index 2ee9dbf2a9..94a996358b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,7 +5,7 @@ Pillow>=7.2 torchvision>=0.8 # lower to 0.7 after PT 1.6 transformers>=4.0 pytorch-tabnet==3.1.1 -datasets>=1.2 +datasets>=1.2, <1.3 pandas>=1.1 scikit-learn>=0.24 numpy # comes with 3rd-party dependency