diff --git a/setup.py b/setup.py index c2c809ad..5757deb3 100644 --- a/setup.py +++ b/setup.py @@ -19,11 +19,11 @@ keras_deps = [ 'tensorflow>=2.2,<2.8', - 'tensorflow_datasets>=4.2,<4.5', + 'tensorflow_datasets>=4.2,<4.6', 'tensorflow-privacy>=0.5,<0.8', ] other_deps = [ - 'pandas~=1.1.0', + 'pandas>=1.1,<1.5', 'scikit-learn>=0.23,<1.1', ] pytorch_deps = [ @@ -45,7 +45,7 @@ grpc_deps = ['grpcio>=1.35,<1.43', 'grpcio-tools>=1.35,<1.44', - 'prometheus_client==0.12.0', + 'prometheus_client==0.13.1', 'click' ] all_deps = list(set(keras_deps + other_deps + pytorch_deps + grpc_deps)) + ["xgboost"]