diff --git a/flaml/automl/spark/__init__.py b/flaml/automl/spark/__init__.py index c8c8111f16..e348627a4b 100644 --- a/flaml/automl/spark/__init__.py +++ b/flaml/automl/spark/__init__.py @@ -11,6 +11,9 @@ from pyspark.pandas import set_option from pyspark.sql import DataFrame as sparkDataFrame from pyspark.util import VersionUtils + import logging + + logger = logging.getLogger(__name__) except ImportError: class psDataFrame: @@ -31,4 +34,5 @@ class psDataFrame: import pandas as pd from pandas import DataFrame, Series except ImportError: + logger.warning("Pandas is not installed. Please install pandas to use DataFrame and Series functionalities.") DataFrame = Series = pd = None