Make sure numpy >= 1.16.0 is installed for fast pickling support (#6486)

* Make sure numpy >= 1.16.0 is installed

* Works for 1.15.4

* lint

* formatting

* update

* put check into the right place

* lint
This commit is contained in:
Philipp Moritz
2019-12-14 16:36:49 -08:00
committed by Edward Oakes
parent ac8f8143e7
commit afae8406da
3 changed files with 14 additions and 1 deletions
+10
View File
@@ -4,6 +4,9 @@ from __future__ import print_function
import logging
import numpy as np
from packaging import version
import ray.ray_constants as ray_constants
@@ -207,3 +210,10 @@ class RayParams(object):
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.")