From 4af42d5bb6ab3fef7dad80c722249218bb9cc061 Mon Sep 17 00:00:00 2001 From: Devin Petersohn Date: Tue, 6 Mar 2018 09:57:49 -0800 Subject: [PATCH] [DataFrame] Adding error checking for pandas version (#1662) * Adding error checking for pandas version * Addressing comments --- python/ray/dataframe/__init__.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/python/ray/dataframe/__init__.py b/python/ray/dataframe/__init__.py index 0bce77403..824ef367a 100644 --- a/python/ray/dataframe/__init__.py +++ b/python/ray/dataframe/__init__.py @@ -1,8 +1,18 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function + +import pandas as pd import threading +pd_version = pd.__version__ +pd_major = int(pd_version.split(".")[0]) +pd_minor = int(pd_version.split(".")[1]) + +if pd_major == 0 and pd_minor < 22: + raise Exception("In order to use Pandas on Ray, please upgrade your Pandas" + " version to >= 0.22.") + DEFAULT_NPARTITIONS = 4