From 408a43fe906f1ff0871ed094974603fb03b7b134 Mon Sep 17 00:00:00 2001 From: Andy Grove Date: Sat, 14 Dec 2024 12:07:29 -0700 Subject: [PATCH] rename python function to match rust --- datafusion_ray/context.py | 2 +- src/query_stage.rs | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/datafusion_ray/context.py b/datafusion_ray/context.py index eedfbe8..8d354ff 100644 --- a/datafusion_ray/context.py +++ b/datafusion_ray/context.py @@ -50,7 +50,7 @@ def execute_query_stage( # if the query stage has a single output partition then we need to execute for the output # partition, otherwise we need to execute in parallel for each input partition - concurrency = stage.get_input_partition_count() + concurrency = stage.get_execution_partition_count() output_partitions_count = stage.get_output_partition_count() if output_partitions_count == 1: # reduce stage diff --git a/src/query_stage.rs b/src/query_stage.rs index a13c701..a5c9a08 100644 --- a/src/query_stage.rs +++ b/src/query_stage.rs @@ -60,7 +60,7 @@ impl PyQueryStage { self.stage.get_child_stage_ids() } - pub fn get_input_partition_count(&self) -> usize { + pub fn get_execution_partition_count(&self) -> usize { self.stage.get_execution_partition_count() }