diff --git a/python/setup.py b/python/setup.py index 2110c2839ca70..ddd961d0412b5 100755 --- a/python/setup.py +++ b/python/setup.py @@ -307,17 +307,17 @@ def run(self): # if you're updating the versions or dependencies. install_requires=["py4j==0.10.9.7"], extras_require={ - "ml": ["numpy>=1.15"], - "mllib": ["numpy>=1.15"], + "ml": ["numpy>=1.15,<2"], + "mllib": ["numpy>=1.15,<2"], "sql": [ "pandas>=%s" % _minimum_pandas_version, "pyarrow>=%s" % _minimum_pyarrow_version, - "numpy>=1.15", + "numpy>=1.15,<2", ], "pandas_on_spark": [ "pandas>=%s" % _minimum_pandas_version, "pyarrow>=%s" % _minimum_pyarrow_version, - "numpy>=1.15", + "numpy>=1.15,<2", ], "connect": [ "pandas>=%s" % _minimum_pandas_version, @@ -325,7 +325,7 @@ def run(self): "grpcio>=%s" % _minimum_grpc_version, "grpcio-status>=%s" % _minimum_grpc_version, "googleapis-common-protos>=%s" % _minimum_googleapis_common_protos_version, - "numpy>=1.15", + "numpy>=1.15,<2", ], }, python_requires=">=3.8",