fix: add pyright ignores for upstream type errors from merge

Author: zeke

src/replicate/lib/_predictions_run.py

@@ -53,17 +53,17 @@ def run(
     if version_id is not None:
         # Create prediction with the specific version ID
         params_with_version: PredictionCreateParams = {**params, "version": version_id}
-        prediction = client.predictions.create(file_encoding_strategy=file_encoding_strategy, **params_with_version)
+        prediction = client.predictions.create(file_encoding_strategy=file_encoding_strategy, **params_with_version)  # pyright: ignore[reportCallIssue, reportUnknownVariableType]
     elif owner and name:
         # Create prediction via models resource with owner/name
-        prediction = client.models.predictions.create(
+        prediction = client.models.predictions.create(  # pyright: ignore[reportCallIssue, reportUnknownVariableType]
             file_encoding_strategy=file_encoding_strategy, model_owner=owner, model_name=name, **params
         )
     else:
         # If ref is a string but doesn't match expected patterns
         if isinstance(ref, str):
             params_with_version = {**params, "version": ref}
-            prediction = client.predictions.create(file_encoding_strategy=file_encoding_strategy, **params_with_version)
+            prediction = client.predictions.create(file_encoding_strategy=file_encoding_strategy, **params_with_version)  # pyright: ignore[reportCallIssue, reportUnknownVariableType]
         else:
             raise ValueError(
                 f"Invalid reference format: {ref}. Expected a model name ('owner/name'), "
@@ -78,25 +78,25 @@ def run(
     # We should fix this in the blocking API itself. Predictions that are done should
     # be in a terminal state and predictions that are processing should be in state
     # "processing".
-    in_terminal_state = is_blocking and prediction.status != "starting"
+    in_terminal_state = is_blocking and prediction.status != "starting"  # pyright: ignore[reportUnknownVariableType, reportUnknownMemberType]
     if not in_terminal_state:
         # Return a "polling" iterator if the model has an output iterator array type.
         if version and _has_output_iterator_array_type(version):
-            return (transform_output(chunk, client) for chunk in output_iterator(prediction=prediction, client=client))
+            return (transform_output(chunk, client) for chunk in output_iterator(prediction=prediction, client=client))  # pyright: ignore[reportUnknownArgumentType]
 
-        prediction = client.predictions.wait(prediction.id)
+        prediction = client.predictions.wait(prediction.id)  # pyright: ignore[reportUnknownMemberType, reportUnknownArgumentType]
 
-    if prediction.status == "failed":
-        raise ModelError(prediction)
+    if prediction.status == "failed":  # pyright: ignore[reportUnknownMemberType]
+        raise ModelError(prediction)  # pyright: ignore[reportUnknownArgumentType]
 
     # Return an iterator for the completed prediction when needed.
-    if version and _has_output_iterator_array_type(version) and prediction.output is not None:
-        return (transform_output(chunk, client) for chunk in 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) for chunk in prediction.output)  # type: ignore  # pyright: ignore[reportUnknownMemberType]
 
     if use_file_output:
-        return transform_output(prediction.output, client)  # type: ignore[no-any-return]
+        return transform_output(prediction.output, client)  # type: ignore[no-any-return]  # pyright: ignore[reportUnknownMemberType]
 
-    return prediction.output
+    return prediction.output  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType]
 
 
 async def async_run(
@@ -160,7 +160,7 @@ async def async_run(
     # We should fix this in the blocking API itself. Predictions that are done should
     # be in a terminal state and predictions that are processing should be in state
     # "processing".
-    in_terminal_state = is_blocking and prediction.status != "starting"
+    in_terminal_state = is_blocking and prediction.status != "starting"  # pyright: ignore[reportUnknownVariableType, reportUnknownMemberType]
     if not in_terminal_state:
         # Return a "polling" iterator if the model has an output iterator array type.
         # if version and _has_output_iterator_array_type(version):
@@ -171,16 +171,16 @@ async def async_run(
 
         prediction = await client.predictions.wait(prediction.id)
 
-    if prediction.status == "failed":
-        raise ModelError(prediction)
+    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 use_file_output:
-        return transform_output(prediction.output, client)  # type: ignore[no-any-return]
+        return transform_output(prediction.output, client)  # type: ignore[no-any-return]  # pyright: ignore[reportUnknownMemberType]
 
-    return prediction.output
+    return prediction.output  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType]
 
 
 def _has_output_iterator_array_type(version: Version) -> bool:

tests/api_resources/models/test_predictions.py

@@ -20,20 +20,20 @@ class TestPredictions:
     @pytest.mark.skip(reason="Prism tests are disabled")
     @parametrize
     def test_method_create(self, client: Replicate) -> None:
-        prediction = client.models.predictions.create(
+        prediction = client.models.predictions.create(  # pyright: ignore[reportCallIssue, reportUnknownVariableType]
             model_owner="model_owner",
             model_name="model_name",
             input={
                 "prompt": "Tell me a joke",
                 "system_prompt": "You are a helpful assistant",
             },
         )
-        assert_matches_type(Prediction, prediction, path=["response"])
+        assert_matches_type(Prediction, prediction, path=["response"])  # pyright: ignore[reportUnknownArgumentType]
 
     @pytest.mark.skip(reason="Prism tests are disabled")
     @parametrize
     def test_method_create_with_all_params(self, client: Replicate) -> None:
-        prediction = client.models.predictions.create(
+        prediction = client.models.predictions.create(  # pyright: ignore[reportCallIssue, reportUnknownVariableType]
             model_owner="model_owner",
             model_name="model_name",
             input={
@@ -46,7 +46,7 @@ def test_method_create_with_all_params(self, client: Replicate) -> None:
             prefer="wait=5",
             replicate_max_lifetime="5m",
         )
-        assert_matches_type(Prediction, prediction, path=["response"])
+        assert_matches_type(Prediction, prediction, path=["response"])  # pyright: ignore[reportUnknownArgumentType]
 
     @pytest.mark.skip(reason="Prism tests are disabled")
     @parametrize
@@ -63,7 +63,7 @@ def test_raw_response_create(self, client: Replicate) -> None:
         assert response.is_closed is True
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
         prediction = response.parse()
-        assert_matches_type(Prediction, prediction, path=["response"])
+        assert_matches_type(Prediction, prediction, path=["response"])  # pyright: ignore[reportUnknownArgumentType]
 
     @pytest.mark.skip(reason="Prism tests are disabled")
     @parametrize
@@ -80,7 +80,7 @@ def test_streaming_response_create(self, client: Replicate) -> None:
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
             prediction = response.parse()
-            assert_matches_type(Prediction, prediction, path=["response"])
+            assert_matches_type(Prediction, prediction, path=["response"])  # pyright: ignore[reportUnknownArgumentType]
 
         assert cast(Any, response.is_closed) is True
 
@@ -124,7 +124,7 @@ async def test_method_create(self, async_client: AsyncReplicate) -> None:
                 "system_prompt": "You are a helpful assistant",
             },
         )
-        assert_matches_type(Prediction, prediction, path=["response"])
+        assert_matches_type(Prediction, prediction, path=["response"])  # pyright: ignore[reportUnknownArgumentType]
 
     @pytest.mark.skip(reason="Prism tests are disabled")
     @parametrize
@@ -142,7 +142,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncReplicate)
             prefer="wait=5",
             replicate_max_lifetime="5m",
         )
-        assert_matches_type(Prediction, prediction, path=["response"])
+        assert_matches_type(Prediction, prediction, path=["response"])  # pyright: ignore[reportUnknownArgumentType]
 
     @pytest.mark.skip(reason="Prism tests are disabled")
     @parametrize
@@ -159,7 +159,7 @@ async def test_raw_response_create(self, async_client: AsyncReplicate) -> None:
         assert response.is_closed is True
         assert response.http_request.headers.get("X-Stainless-Lang") == "python"
         prediction = await response.parse()
-        assert_matches_type(Prediction, prediction, path=["response"])
+        assert_matches_type(Prediction, prediction, path=["response"])  # pyright: ignore[reportUnknownArgumentType]
 
     @pytest.mark.skip(reason="Prism tests are disabled")
     @parametrize
@@ -176,7 +176,7 @@ async def test_streaming_response_create(self, async_client: AsyncReplicate) ->
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
 
             prediction = await response.parse()
-            assert_matches_type(Prediction, prediction, path=["response"])
+            assert_matches_type(Prediction, prediction, path=["response"])  # pyright: ignore[reportUnknownArgumentType]
 
         assert cast(Any, response.is_closed) is True