From 32f275344bd5d2034d1cb2a5384b204d485ecde3 Mon Sep 17 00:00:00 2001
From: Richard Liaw <rliaw@berkeley.edu>
Date: Mon, 5 Aug 2019 18:59:08 -0700
Subject: [PATCH] [tune] Pandas as soft dep

---
 python/ray/tune/analysis/experiment_analysis.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/python/ray/tune/analysis/experiment_analysis.py b/python/ray/tune/analysis/experiment_analysis.py
index e8361222c..1db7f1f9e 100644
--- a/python/ray/tune/analysis/experiment_analysis.py
+++ b/python/ray/tune/analysis/experiment_analysis.py
@@ -6,7 +6,11 @@ import copy
 import json
 import logging
 import os
-import pandas as pd
+
+try:
+    import pandas as pd
+except ImportError:
+    pd = None
 
 from ray.tune.error import TuneError
 from ray.tune.result import EXPR_PROGRESS_FILE, EXPR_PARAM_FILE
@@ -45,7 +49,13 @@ class Analysis(object):
         self._experiment_dir = experiment_dir
         self._configs = {}
         self._trial_dataframes = {}
-        self.fetch_trial_dataframes()
+
+        if not pd:
+            logger.warning(
+                "pandas not installed. Run `pip install pandas` for "
+                "Analysis utilities.")
+        else:
+            self.fetch_trial_dataframes()
 
     def dataframe(self, metric=None, mode=None):
         """Returns a pandas.DataFrame object constructed from the trials.