diff --git a/setup.py b/setup.py index a3be04668..876b25d60 100644 --- a/setup.py +++ b/setup.py @@ -27,7 +27,7 @@ DOCLINES = (__doc__ or "").split("\n") FRAMEWORKS = ["tensorflow", "pytorch", "mxnet", "xgboost"] TESTS_PACKAGES = ["pytest", "torchvision", "pandas"] -INSTALL_REQUIRES = ["protobuf>=3.6.0", "numpy", "packaging", "boto3>=1.10.32"] +INSTALL_REQUIRES = ["protobuf>=3.6.0", "numpy>1.16.0,<2.0.0", "packaging", "boto3>=1.10.32"] def compile_summary_protobuf():