@@ -239,10 +239,9 @@ class ModelSlice(Slice):
239
239
@classmethod
240
240
def query_str (cls ):
241
241
query_str = """
242
- query getDataRowIdenfifiersBySavedModelQueryPyApi($id: ID!, $modelRunId: ID, $ from: DataRowIdentifierCursorInput, $first: Int!) {
242
+ query getDataRowIdenfifiersBySavedModelQueryPyApi($id: ID!, $from: DataRowIdentifierCursorInput, $first: Int!) {
243
243
getDataRowIdentifiersBySavedModelQuery(input: {
244
244
savedQueryId: $id,
245
- modelRunId: $modelRunId,
246
245
after: $from
247
246
first: $first
248
247
}) {
@@ -264,23 +263,17 @@ def query_str(cls):
264
263
"""
265
264
return query_str
266
265
267
- def get_data_row_ids (self , model_run_id : str ) -> PaginatedCollection :
266
+ def get_data_row_ids (self ) -> PaginatedCollection :
268
267
"""
269
268
Fetches all data row ids that match this Slice
270
269
271
- Params
272
- model_run_id: str, required, uid or cuid of model run
273
-
274
270
Returns:
275
271
A PaginatedCollection of data row ids
276
272
"""
277
273
return PaginatedCollection (
278
274
client = self .client ,
279
275
query = ModelSlice .query_str (),
280
- params = {
281
- 'id' : str (self .uid ),
282
- 'modelRunId' : model_run_id
283
- },
276
+ params = {'id' : str (self .uid )},
284
277
dereferencing = ['getDataRowIdentifiersBySavedModelQuery' , 'nodes' ],
285
278
obj_class = lambda _ , data_row_id_and_gk : data_row_id_and_gk .get ('id'
286
279
),
@@ -289,24 +282,17 @@ def get_data_row_ids(self, model_run_id: str) -> PaginatedCollection:
289
282
'endCursor'
290
283
])
291
284
292
- def get_data_row_identifiers (self ,
293
- model_run_id : str ) -> PaginatedCollection :
285
+ def get_data_row_identifiers (self ) -> PaginatedCollection :
294
286
"""
295
287
Fetches all data row ids and global keys (where defined) that match this Slice
296
288
297
- Params:
298
- model_run_id : str, required, uid or cuid of model run
299
-
300
289
Returns:
301
290
A PaginatedCollection of Slice.DataRowIdAndGlobalKey
302
291
"""
303
292
return PaginatedCollection (
304
293
client = self .client ,
305
294
query = ModelSlice .query_str (),
306
- params = {
307
- 'id' : str (self .uid ),
308
- 'modelRunId' : model_run_id
309
- },
295
+ params = {'id' : str (self .uid )},
310
296
dereferencing = ['getDataRowIdentifiersBySavedModelQuery' , 'nodes' ],
311
297
obj_class = lambda _ , data_row_id_and_gk : Slice .DataRowIdAndGlobalKey (
312
298
data_row_id_and_gk .get ('id' ),
0 commit comments