Skip to content
Closed
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix
  • Loading branch information
grundprinzip committed Mar 29, 2024
commit 5687f6ce8e718987657b36de8a45a2c004f76ec2
4 changes: 3 additions & 1 deletion python/pyspark/sql/connect/shell/progress.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,9 @@ def __call__(
pass


def from_proto(proto: ExecutePlanResponse) -> typing.Tuple[Iterable[StageInfo], int]:
def from_proto(
proto: ExecutePlanResponse.ExecutionProgress,
) -> typing.Tuple[Iterable[StageInfo], int]:
result = []
for stage in proto.stages:
result.append(
Expand Down