(keys)
| 1283 | |
| 1284 | |
| 1285 | def _convert_dask_keys(keys): |
| 1286 | from dask._task_spec import List, TaskRef |
| 1287 | |
| 1288 | assert isinstance(keys, list) |
| 1289 | new_keys = [] |
| 1290 | for key in keys: |
| 1291 | if isinstance(key, list): |
| 1292 | new_keys.append(_convert_dask_keys(key)) |
| 1293 | else: |
| 1294 | new_keys.append(TaskRef(key)) |
| 1295 | return List(*new_keys) |
| 1296 | |
| 1297 | |
| 1298 | class HLGFinalizeCompute(HLGExpr): |