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.