def _rf_init(spark_session: SparkSession) -> SparkSession:
""" Adds RasterFrames functionality to PySpark session."""
if not hasattr(spark_session, "rasterframes"):
spark_session.rasterframes = RFContext(spark_session)
spark_session.sparkContext._rf_context = spark_session.rasterframes
return spark_session
