Skip to content

Commit

Permalink
remove unused code
Browse files Browse the repository at this point in the history
  • Loading branch information
universalmind303 committed Jan 15, 2025
1 parent d5766af commit 5d82992
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 5 deletions.
4 changes: 0 additions & 4 deletions daft/daft/__init__.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -1689,10 +1689,6 @@ class LogicalPlanBuilder:

class NativeExecutor:
def __init__(self) -> None: ...
# @staticmethod
# def from_logical_plan_builder(
# logical_plan_builder: LogicalPlanBuilder,
# ) -> NativeExecutor: ...
def run(
self, psets: dict[str, list[PartitionT]], cfg: PyDaftExecutionConfig, results_buffer_size: int | None
) -> Iterator[PyMicroPartition]: ...
Expand Down
3 changes: 2 additions & 1 deletion daft/runners/pyrunner.py
Original file line number Diff line number Diff line change
Expand Up @@ -380,8 +380,9 @@ def run_iter(
if daft_execution_config.enable_native_executor:
logger.info("Using native executor")

executor = NativeExecutor.from_logical_plan_builder(builder)
executor = NativeExecutor()
results_gen = executor.run(
builder,
{k: v.values() for k, v in self._part_set_cache.get_all_partition_sets().items()},
daft_execution_config,
results_buffer_size,
Expand Down

0 comments on commit 5d82992

Please sign in to comment.