Make sure numpy >= 1.16.0 is installed for fast pickling support (ray…

…-project#6486)

* Make sure numpy >= 1.16.0 is installed

* Works for 1.15.4

* lint

* formatting

* update

* put check into the right place

* lint

python/ray/parameter.py

@@ -4,6 +4,9 @@
 
 import logging
 
+import numpy as np
+from packaging import version
+
 import ray.ray_constants as ray_constants
 
 
@@ -207,3 +210,10 @@ def _check_usage(self):
         if self.redirect_output is not None:
             raise DeprecationWarning(
                 "The redirect_output argument is deprecated.")
+
+        if self.use_pickle:
+            assert (version.parse(
+                np.__version__) >= version.parse("1.16.0")), (
+                    "numpy >= 1.16.0 required for use_pickle=True support. "
+                    "You can use ray.init(use_pickle=False) for older numpy "
+                    "versions, but this may be removed in future versions.")

python/ray/serialization.py

@@ -1,12 +1,15 @@
 from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
+
 import hashlib
 import io
 import logging
 import time
+
 import pyarrow
 import pyarrow.plasma as plasma
+
 import ray.cloudpickle as pickle
 from ray import ray_constants, JobID
 import ray.utils

python/setup.py

@@ -160,7 +160,7 @@ def find_version(*filepath):
 
 
 requires = [
-    "numpy >= 1.14",
+    "numpy >= 1.16",
     "filelock",
     "jsonschema",
     "funcsigs",