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.

Sign up for GitHub

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

Jump to bottom

Add Python wrapper for LogicalPlan::Union #240

Merged
merged 2 commits into from
Feb 28, 2023
Merged

Add Python wrapper for LogicalPlan::Union #240

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
d4cb59d
add union
iajoiner Feb 26, 2023
449557e
Make clippy happier
iajoiner Feb 26, 2023
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
Make clippy happier
  • Loading branch information
@iajoiner
iajoiner committed Feb 26, 2023
commit 449557e494074e0be1de34c71ac35c49f794f79f
2 changes: 1 addition & 1 deletion src/context.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -605,7 +605,7 @@ impl PySessionContext {
let plan = plan.plan.clone();
let fut: JoinHandle<datafusion_common::Result<SendableRecordBatchStream>> =
rt.spawn(async move { plan.execute(part, ctx) });
let stream = wait_for_future(py, fut).map_err(|e| py_datafusion_err(e))?;
let stream = wait_for_future(py, fut).map_err(py_datafusion_err)?;
Ok(PyRecordBatchStream::new(stream?))
}
}
Expand Down
3 changes: 1 addition & 2 deletions src/expr/union.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -76,8 +76,7 @@ impl PyUnion {

impl LogicalNode for PyUnion {
fn inputs(&self) -> Vec<PyLogicalPlan> {
(*self)
.union_
self.union_
.inputs
.iter()
.map(|x| x.as_ref().clone().into())
Expand Down