Skip to content

Commit 1b86d4a

Browse files
committed
batch: fix Ray runner types
1 parent 833be0c commit 1b86d4a

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/lenskit/batch/_ray.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
from lenskit.parallel.ray import TaskLimiter, ensure_cluster
1919
from lenskit.pipeline import Pipeline, ProfileSink
2020

21-
from ._queries import BatchRequest
21+
from ._queries import ResolvedBatchRequest
2222
from ._results import BatchResultRow
2323
from ._runner import InvocationSpec, run_pipeline
2424

@@ -29,7 +29,7 @@ def ray_results(
2929
pipeline: Pipeline,
3030
profiler: ProfileSink | None,
3131
invocations: list[InvocationSpec],
32-
queries: Iterable[BatchRequest],
32+
queries: Iterable[ResolvedBatchRequest],
3333
n_jobs: int,
3434
batch_size: int,
3535
) -> Generator[BatchResultRow]:
@@ -58,6 +58,6 @@ def _run_batch(
5858
pipeline: Pipeline,
5959
invocations: list[InvocationSpec],
6060
profiler: ProfileSink | None,
61-
requests: Sequence[BatchRequest],
61+
requests: Sequence[ResolvedBatchRequest],
6262
) -> list[BatchResultRow]:
6363
return [run_pipeline(pipeline, invocations, profiler, req) for req in requests]

0 commit comments

Comments
 (0)