# © Copyright Databand.ai, an IBM Company 2022

# we should run bootstrap first! before we import any other pyspark object
from dbnd_spark.spark_bootstrap import dbnd_spark_bootstrap


dbnd_spark_bootstrap()

__all__ = []

try:
    from dbnd_spark.spark import PySparkTask, SparkTask, spark_task
    from dbnd_spark.spark_config import SparkConfig
    from dbnd_spark.spark_session import get_spark_session

    __all__ += [
        "PySparkTask",
        "SparkTask",
        "spark_task",
        "SparkConfig",
        "get_spark_session",
    ]
except ImportError:
    raise
