There was an error while loading. Please reload this page.
1 parent 833be0c commit 1b86d4aCopy full SHA for 1b86d4a
1 file changed
src/lenskit/batch/_ray.py
@@ -18,7 +18,7 @@
18
from lenskit.parallel.ray import TaskLimiter, ensure_cluster
19
from lenskit.pipeline import Pipeline, ProfileSink
20
21
-from ._queries import BatchRequest
+from ._queries import ResolvedBatchRequest
22
from ._results import BatchResultRow
23
from ._runner import InvocationSpec, run_pipeline
24
@@ -29,7 +29,7 @@ def ray_results(
29
pipeline: Pipeline,
30
profiler: ProfileSink | None,
31
invocations: list[InvocationSpec],
32
- queries: Iterable[BatchRequest],
+ queries: Iterable[ResolvedBatchRequest],
33
n_jobs: int,
34
batch_size: int,
35
) -> Generator[BatchResultRow]:
@@ -58,6 +58,6 @@ def _run_batch(
58
59
60
61
- requests: Sequence[BatchRequest],
+ requests: Sequence[ResolvedBatchRequest],
62
) -> list[BatchResultRow]:
63
return [run_pipeline(pipeline, invocations, profiler, req) for req in requests]
0 commit comments