(keys)
| 1320 | |
| 1321 | |
| 1322 | def _convert_dask_keys(keys): |
| 1323 | from dask._task_spec import List, TaskRef |
| 1324 | |
| 1325 | assert isinstance(keys, list) |
| 1326 | new_keys = [] |
| 1327 | for key in keys: |
| 1328 | if isinstance(key, list): |
| 1329 | new_keys.append(_convert_dask_keys(key)) |
| 1330 | else: |
| 1331 | new_keys.append(TaskRef(key)) |
| 1332 | return List(*new_keys) |
| 1333 | |
| 1334 | |
| 1335 | class HLGFinalizeCompute(HLGExpr): |