@@ -263,8 +263,7 @@ def update_status(self,
263263 experimental = True )
264264
265265 @experimental
266- def update_config (self ,
267- config : Dict [str , Any ]):
266+ def update_config (self , config : Dict [str , Any ]):
268267 data : Dict [str , Any ] = {'config' : config }
269268 res = self .client .execute (
270269 """mutation updateModelRunConfigPyApi($modelRunId: ID!, $data: UpdateModelRunConfigInput!){
@@ -275,31 +274,26 @@ def update_config(self,
275274 'data' : data
276275 },
277276 experimental = True )
278- return res ["updateModelRunConfig" ]
277+ return res ["updateModelRunConfig" ]
279278
280279 @experimental
281280 def reset_config (self ):
282281 res = self .client .execute (
283282 """mutation resetModelRunConfigPyApi($modelRunId: ID!){
284283 resetModelRunConfig(modelRun: {id : $modelRunId}){trainingMetadata}
285284 }
286- """ , {
287- 'modelRunId' : self .uid
288- },
285+ """ , {'modelRunId' : self .uid },
289286 experimental = True )
290- return res ["resetModelRunConfig" ]
291-
287+ return res ["resetModelRunConfig" ]
288+
292289 @experimental
293290 def fetch_config (self ):
294- res = self .client .execute (
295- """query ModelRunPyApi($modelRunId: ID!){
291+ res = self .client .execute ("""query ModelRunPyApi($modelRunId: ID!){
296292 modelRun(where: {id : $modelRunId}){trainingMetadata}
297293 }
298- """ , {
299- 'modelRunId' : self .uid
300- },
301- experimental = True )
302- return res ["modelRun" ]
294+ """ , {'modelRunId' : self .uid },
295+ experimental = True )
296+ return res ["modelRun" ]
303297
304298 @experimental
305299 def export_labels (
0 commit comments