@@ -164,7 +164,8 @@ def run(
164
164
self ,
165
165
ref : str ,
166
166
input : Optional [Dict [str , Any ]] = None ,
167
- use_file_output : Optional [bool ] = None ,
167
+ * ,
168
+ use_file_output : Optional [bool ] = True ,
168
169
** params : Unpack ["Predictions.CreatePredictionParams" ],
169
170
) -> Union [Any , Iterator [Any ]]: # noqa: ANN401
170
171
"""
@@ -177,7 +178,8 @@ async def async_run(
177
178
self ,
178
179
ref : str ,
179
180
input : Optional [Dict [str , Any ]] = None ,
180
- use_file_output : Optional [bool ] = None ,
181
+ * ,
182
+ use_file_output : Optional [bool ] = True ,
181
183
** params : Unpack ["Predictions.CreatePredictionParams" ],
182
184
) -> Union [Any , AsyncIterator [Any ]]: # noqa: ANN401
183
185
"""
@@ -191,28 +193,30 @@ async def async_run(
191
193
def stream (
192
194
self ,
193
195
ref : str ,
196
+ * ,
194
197
input : Optional [Dict [str , Any ]] = None ,
195
- use_file_output : Optional [bool ] = None ,
198
+ use_file_output : Optional [bool ] = True ,
196
199
** params : Unpack ["Predictions.CreatePredictionParams" ],
197
200
) -> Iterator ["ServerSentEvent" ]:
198
201
"""
199
202
Stream a model's output.
200
203
"""
201
204
202
- return stream (self , ref , input , use_file_output , ** params )
205
+ return stream (self , ref , input , use_file_output = use_file_output , ** params )
203
206
204
207
async def async_stream (
205
208
self ,
206
209
ref : str ,
207
210
input : Optional [Dict [str , Any ]] = None ,
208
- use_file_output : Optional [bool ] = None ,
211
+ * ,
212
+ use_file_output : Optional [bool ] = True ,
209
213
** params : Unpack ["Predictions.CreatePredictionParams" ],
210
214
) -> AsyncIterator ["ServerSentEvent" ]:
211
215
"""
212
216
Stream a model's output asynchronously.
213
217
"""
214
218
215
- return async_stream (self , ref , input , use_file_output , ** params )
219
+ return async_stream (self , ref , input , use_file_output = use_file_output , ** params )
216
220
217
221
218
222
# Adapted from https://github.com/encode/httpx/issues/108#issuecomment-1132753155
0 commit comments