Skip to content

Commit 6f10116

Browse files
committed
fix(predictions): use Omit instead of NotGiven
1 parent d1bebb6 commit 6f10116

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

src/replicate/resources/models/predictions.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
from replicate.lib._files import FileEncodingStrategy, encode_json, async_encode_json
1111

12-
from ..._types import Body, Query, Headers, NotGiven, not_given
12+
from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
1313
from ..._utils import maybe_transform, strip_not_given, async_maybe_transform
1414
from ..._compat import cached_property
1515
from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -52,10 +52,10 @@ def create(
5252
model_owner: str,
5353
model_name: str,
5454
input: object,
55-
stream: bool | NotGiven = NOT_GIVEN,
56-
webhook: str | NotGiven = NOT_GIVEN,
57-
webhook_events_filter: List[Literal["start", "output", "logs", "completed"]] | NotGiven = NOT_GIVEN,
58-
prefer: str | NotGiven = NOT_GIVEN,
55+
stream: bool | Omit = omit,
56+
webhook: str | Omit = omit,
57+
webhook_events_filter: List[Literal["start", "output", "logs", "completed"]] | Omit = omit,
58+
prefer: str | Omit = omit,
5959
file_encoding_strategy: Optional["FileEncodingStrategy"] = None,
6060
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
6161
# The extra values given here take precedence over values defined on the client or passed to this method.
@@ -214,10 +214,10 @@ async def create(
214214
model_owner: str,
215215
model_name: str,
216216
input: object,
217-
stream: bool | NotGiven = NOT_GIVEN,
218-
webhook: str | NotGiven = NOT_GIVEN,
219-
webhook_events_filter: List[Literal["start", "output", "logs", "completed"]] | NotGiven = NOT_GIVEN,
220-
prefer: str | NotGiven = NOT_GIVEN,
217+
stream: bool | Omit = omit,
218+
webhook: str | Omit = omit,
219+
webhook_events_filter: List[Literal["start", "output", "logs", "completed"]] | Omit = omit,
220+
prefer: str | Omit = omit,
221221
file_encoding_strategy: Optional["FileEncodingStrategy"] = None,
222222
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
223223
# The extra values given here take precedence over values defined on the client or passed to this method.

0 commit comments

Comments
 (0)