The whole idea was to introduce the internal koheesio spark session to be able to provide easy switch between remote and local modes.

Also if I'm not wrong pydantic is checking SparkSession type based on the full import path and with remote spark session it is imported from a different path, at least it was in a such way sometime ago

https://docs.databricks.com/en/dev-tools/databricks-connect/python/limitations.html

check the affected/reference code within Koheesio

Not available on Databricks Connect for Databricks Runtime 13.3 LTS and below:

• Streaming foreachBatch ==> @riccamini -- NA -- works for 14.3 & above. confirm this assumption.

Not available:

• Databricks Utilities: credentials, library, notebook workflow, widgets -- not sure what does this mean (brickflow is affected ) ==> @pariksheet with @asingamaneni

• SparkContext -- No jvm operations ==> @mikita-sakalouski @dannymeijer

• Changing the log4j log level through SparkContext -- need to check the code ==> Nathan

Run the unit test locally with spark-connect remote instant.

Check how to manage SparkSession and DatabricksSession.

I have added details in here related to foreachbatch function: #56

If you prefer collecting everything here I will copy paste the comment and close the issue.

One additional point that I do not see in the list is Dataframe.rdd which is being used in some tests

• Delta merge through Delta API not possible. (only supported from Spark 4.0 and Delta 4.0)
• Delta Merge through SPARK SQL still works
• How to support 3.x and 4.x spark/delta versions
• Support for Snowflake DDLs through _jvm

There is a way to check Spark Session is remote or native.

we should introduce api/function to get spark session flag and check against the specific APIs e.g. delta merge /snowflake and raise the exception.

-- use snowflake-connector-python instead of spark._jvm

All of these should be addressed as part of release 0.9.0 (currently in pre-release). Please verify your usecases accordingly so we can proceed with the release.