Skip to content

Commit ffdaeda

Browse files
committed
Fix async run()
1 parent 38cdddc commit ffdaeda

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/replicate/_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -326,7 +326,7 @@ async def run(
326326
ref: str,
327327
*,
328328
wait: Union[int, bool, NotGiven] = NOT_GIVEN,
329-
**params: Unpack[PredictionCreateParams],
329+
**params: Unpack[PredictionCreateParamsWithoutVersion],
330330
) -> Any:
331331
"""Run a model and wait for its output."""
332332
from .lib._predictions import async_run

src/replicate/lib/_predictions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ async def async_run(
7575
*,
7676
wait: Union[int, bool, NotGiven] = NOT_GIVEN,
7777
# use_file_output: Optional[bool] = True,
78-
**params: Unpack[PredictionCreateParams],
78+
**params: Unpack[PredictionCreateParamsWithoutVersion],
7979
) -> PredictionOutput | FileOutput | Iterable[FileOutput] | Dict[str, FileOutput]:
8080
from ._files import transform_output
8181

0 commit comments

Comments
 (0)