alamb commented on pull request #9865: URL: https://github.com/apache/arrow/pull/9865#issuecomment-813013550
I spent some time looking at this. It seems as if at least one problem is that the `TopKExec` user defined plan should declare that it takes only a single input partition from its child: ``` #[async_trait] impl ExecutionPlan for TopKExec { ... fn required_child_distribution(&self) -> Distribution { Distribution::SinglePartition } ... ``` However, that is still not enough to get the tests to pass. My debugging of `cargo test -p datafusion --test user_defined_plan` suggests that the actual TopK stream never runs; I'll keep putzing with it -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org