Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix the build error for Spark 400 #11905

Merged
merged 1 commit into from
Dec 30, 2024

Conversation

firestarman
Copy link
Collaborator

@firestarman firestarman commented Dec 25, 2024

close #11895

Spark 400 introduces a new parameter in BasePythonRunner, so this PR creates a new shim for this change and specifies this to an empty map to pass the build. It is ok since this metric parameter is only for debugging.

Signed-off-by: Firestarman <firestarmanllc@gmail.com>
@firestarman
Copy link
Collaborator Author

build

funcs: Seq[ChainedPythonFunctions],
evalType: Int,
argOffsets: Array[Array[Int]],
jobArtifactUUID: Option[String]
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Add a default Parameters for calss GpuBasePythonRunner?
Then user can pass in the map.
like:

abstract class GpuBasePythonRunner[IN](
    funcs: Seq[ChainedPythonFunctions],
    evalType: Int,
    argOffsets: Array[Array[Int]],
    jobArtifactUUID: Option[String],
    cfg: Map[String, String] = Map.empty
) extends BasePythonRunner[IN, ColumnarBatch](funcs, evalType, argOffsets, jobArtifactUUID,
    cfg)

Copy link
Collaborator Author

@firestarman firestarman Dec 25, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We need to change this for all the shims if the signature changes, so for a quick fix, an empty map should be good enough for now.
We will do as your suggestion in following PRs if this becomes a necessary parameter for GPU.

@firestarman firestarman requested a review from res-life December 30, 2024 01:35
@firestarman firestarman merged commit d9b5c53 into NVIDIA:branch-25.02 Dec 30, 2024
50 of 51 checks passed
@firestarman firestarman deleted the py-udf-metric branch December 30, 2024 02:29
@sameerz sameerz added the Spark 4.0+ Spark 4.0+ issues label Jan 3, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Spark 4.0+ Spark 4.0+ issues
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[BUG] BasePythonRunner has a new parameter metrics in Spark 4.0
3 participants