Update run() return type to Any

Author: dgellow

examples/demo.py

@@ -1,8 +1,6 @@
-from typing import Iterable, cast
-
 import rich
 
-from replicate import FileOutput, ReplicateClient
+from replicate import ReplicateClient
 
 client = ReplicateClient()
 
@@ -11,6 +9,6 @@
     input={"prompt": "astronaut riding a rocket like a horse"},
 )
 rich.print(outputs)
-for index, output in enumerate(cast(Iterable[FileOutput], outputs)):
+for index, output in enumerate(outputs):
     with open(f"output_{index}.webp", "wb") as file:
         file.write(output.read())

src/replicate/_client.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 import os
-from typing import TYPE_CHECKING, Any, Dict, Union, Mapping, Iterable
+from typing import TYPE_CHECKING, Any, Union, Mapping
 from typing_extensions import Self, Unpack, override
 
 import httpx
@@ -12,7 +12,7 @@
 
 from . import _exceptions
 from ._qs import Querystring
-from .types import PredictionOutput, PredictionCreateParams
+from .types import PredictionCreateParams
 from ._types import (
     NOT_GIVEN,
     Omit,
@@ -36,9 +36,6 @@
 from .resources.webhooks import webhooks
 from .resources.deployments import deployments
 
-if TYPE_CHECKING:
-    from .lib._files import FileOutput
-
 __all__ = [
     "Timeout",
     "Transport",
@@ -135,7 +132,7 @@ def run(
         *,
         wait: Union[int, bool, NotGiven] = NOT_GIVEN,
         **params: Unpack[PredictionCreateParamsWithoutVersion],
-    ) -> PredictionOutput | FileOutput | Iterable[FileOutput] | Dict[str, FileOutput]:
+    ) -> Any:
         """Run a model and wait for its output."""
         from .lib._predictions import run
 
@@ -330,7 +327,7 @@ async def run(
         *,
         wait: Union[int, bool, NotGiven] = NOT_GIVEN,
         **params: Unpack[PredictionCreateParams],
-    ) -> PredictionOutput | FileOutput | Iterable[FileOutput] | Dict[str, FileOutput]:
+    ) -> Any:
         """Run a model and wait for its output."""
         from .lib._predictions import async_run
 

src/replicate/types/prediction_create_params.py

@@ -95,4 +95,4 @@ class PredictionCreateParamsWithoutVersion(TypedDict, total=False):
 
 class PredictionCreateParams(PredictionCreateParamsWithoutVersion):
     version: Required[str]
-    """The ID of the model version that you want to run."""
+    """The ID of the model version that you want to run."""