Create spark with rikai extension

create-spark-session.py

spark = (
    SparkSession
    .builder
    .config("spark.jars.packages", "ai.eto:rikai_2.12:0.1.0")
    .config(
        "spark.sql.extensions",
        "ai.eto.rikai.sql.spark.RikaiSparkSessionExtensions",
    )
    .config(
        "spark.rikai.sql.ml.registry.mlflow.tracking_uri",
        MLFLOW_TRACKING_URI,
    )
    .master("local[*]")
    .getOrCreate()
);