fix: add remaining pyright ignores for async functions

Author: zeke

src/replicate/lib/_predictions_run.py

@@ -131,19 +131,19 @@ async def async_run(
     if version_id is not None:
         # Create prediction with the specific version ID
         params_with_version: PredictionCreateParams = {**params, "version": version_id}
-        prediction = await client.predictions.create(
+        prediction = await client.predictions.create(  # pyright: ignore[reportCallIssue, reportUnknownVariableType]
             file_encoding_strategy=file_encoding_strategy, **params_with_version
         )
     elif owner and name:
         # Create prediction via models resource with owner/name
-        prediction = await client.models.predictions.create(
+        prediction = await client.models.predictions.create(  # pyright: ignore[reportCallIssue, reportUnknownVariableType]
             model_owner=owner, model_name=name, file_encoding_strategy=file_encoding_strategy, **params
         )
     else:
         # If ref is a string but doesn't match expected patterns
         if isinstance(ref, str):
             params_with_version = {**params, "version": ref}
-            prediction = await client.predictions.create(
+            prediction = await client.predictions.create(  # pyright: ignore[reportCallIssue, reportUnknownVariableType]
                 file_encoding_strategy=file_encoding_strategy, **params_with_version
             )
         else:
@@ -169,14 +169,14 @@ async def async_run(
         #         async for chunk in prediction.async_output_iterator()
         #     )
 
-        prediction = await client.predictions.wait(prediction.id)
+        prediction = await client.predictions.wait(prediction.id)  # pyright: ignore[reportUnknownMemberType, reportUnknownArgumentType]
 
     if prediction.status == "failed":  # pyright: ignore[reportUnknownMemberType]
         raise ModelError(prediction)  # pyright: ignore[reportUnknownArgumentType]
 
     # Return an iterator for completed output if the model has an output iterator array type.
-    if version and _has_output_iterator_array_type(version) and prediction.output is not None:
-        return (transform_output(chunk, client) async for chunk in _make_async_iterator(prediction.output))  # type: ignore
+    if version and _has_output_iterator_array_type(version) and prediction.output is not None:  # pyright: ignore[reportUnknownMemberType]
+        return (transform_output(chunk, client) async for chunk in _make_async_iterator(prediction.output))  # type: ignore  # pyright: ignore[reportUnknownMemberType]
     if use_file_output:
         return transform_output(prediction.output, client)  # type: ignore[no-any-return]  # pyright: ignore[reportUnknownMemberType]
 

tests/api_resources/models/test_predictions.py

@@ -44,7 +44,7 @@ def test_method_create_with_all_params(self, client: Replicate) -> None:
             webhook="https://example.com/my-webhook-handler",
             webhook_events_filter=["start", "completed"],
             prefer="wait=5",
-            replicate_max_lifetime="5m",
+            replicate_max_lifetime="5m",  # pyright: ignore[reportCallIssue]
         )
         assert_matches_type(Prediction, prediction, path=["response"])  # pyright: ignore[reportUnknownArgumentType]
 
@@ -140,7 +140,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncReplicate)
             webhook="https://example.com/my-webhook-handler",
             webhook_events_filter=["start", "completed"],
             prefer="wait=5",
-            replicate_max_lifetime="5m",
+            replicate_max_lifetime="5m",  # pyright: ignore[reportCallIssue]
         )
         assert_matches_type(Prediction, prediction, path=["response"])  # pyright: ignore[reportUnknownArgumentType]
 

tests/api_resources/test_predictions.py

@@ -38,7 +38,7 @@ def test_method_create_with_all_params(self, client: Replicate) -> None:
             webhook="https://example.com/my-webhook-handler",
             webhook_events_filter=["start", "completed"],
             prefer="wait=5",
-            replicate_max_lifetime="5m",
+            replicate_max_lifetime="5m",  # pyright: ignore[reportCallIssue]
         )
         assert_matches_type(Prediction, prediction, path=["response"])  # pyright: ignore[reportUnknownArgumentType]
 
@@ -216,7 +216,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncReplicate)
             webhook="https://example.com/my-webhook-handler",
             webhook_events_filter=["start", "completed"],
             prefer="wait=5",
-            replicate_max_lifetime="5m",
+            replicate_max_lifetime="5m",  # pyright: ignore[reportCallIssue]
         )
         assert_matches_type(Prediction, prediction, path=["response"])  # pyright: ignore[reportUnknownArgumentType]