Skip to content

Support spark connect#62

Merged
WeichenXu123 merged 13 commits into
joblib:masterfrom
WeichenXu123:spark-connect
Apr 7, 2025
Merged

Support spark connect#62
WeichenXu123 merged 13 commits into
joblib:masterfrom
WeichenXu123:spark-connect

Conversation

@WeichenXu123
Copy link
Copy Markdown
Collaborator

@WeichenXu123 WeichenXu123 commented Mar 26, 2025

Support spark connect:

  • Replace sc.parallelize with dataframe.mapInPandas
  • Replace setJobGroup, cancelJobGroup with setTag, cancelTag
  • Set resource profile through mapInPandas
  • Patch inheritable_thread_target to propagate SparkSession instance to thread locals.
  • Remove CI tasks against old Spark versions.

Test notebook:
https://e2-dogfood.staging.cloud.databricks.com/editor/notebooks/4292352991115136?o=6051921418418893#command/7732762698274238

Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
@WeichenXu123 WeichenXu123 changed the title [WIP] Support spark connect Support spark connect Apr 7, 2025
Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
Comment thread test/test_backend.py
Comment thread .github/workflows/main.yml Outdated
Comment thread joblibspark/backend.py Outdated
Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
@WeichenXu123 WeichenXu123 merged commit 8edfb10 into joblib:master Apr 7, 2025
8 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants