[Dataset] Bump numpy >=1.20 dependency (ray-project#20374)

* done? * . Co-authored-by: Alex Wu <[email protected]>
charleslai2000 · Nov 15, 2021 · 884bb3d · 884bb3d
1 parent d191ad2
commit 884bb3d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/docker/base-deps/Dockerfile b/docker/base-deps/Dockerfile
@@ -53,7 +53,8 @@ RUN sudo apt-get update -y && sudo apt-get upgrade -y \
  && $HOME/anaconda3/bin/pip install --no-cache-dir \
  flatbuffers \
  cython==0.29.23 \
- numpy==1.19.5 \
+ # Necessary for Dataset to work properly.
+ numpy>=1.20 \
  psutil \
  blist \ 
  # blist is needed for numpy (which is re-installed when ray is installed)

diff --git a/python/setup.py b/python/setup.py
@@ -219,6 +219,10 @@ def get_packages(self):
  ],
  }
 
+ if sys.version_info >= (3, 7):
+ # Numpy dropped python 3.6 support in 1.20.
+ setup_spec.extras["data"].append("numpy >= 1.20")
+
  # Ray Serve depends on the Ray dashboard components.
  setup_spec.extras["serve"] = list(
  set(setup_spec.extras["serve"] + setup_spec.extras["default"]))