(*args, **kwargs)
| 377 | |
| 378 | |
| 379 | def _materialize_futures(*args, **kwargs): |
| 380 | new_args = [] |
| 381 | for arg in args: |
| 382 | if isinstance(arg, DataProtoFuture): |
| 383 | arg = arg.get() |
| 384 | # add more type to materialize |
| 385 | new_args.append(arg) |
| 386 | for k, v in kwargs.items(): |
| 387 | if isinstance(v, DataProtoFuture): |
| 388 | kwargs[k] = v.get() |
| 389 | |
| 390 | new_args = tuple(new_args) |
| 391 | return new_args, kwargs |
| 392 | |
| 393 | |
| 394 | def register(dispatch_mode=Dispatch.ALL_TO_ALL, execute_mode=Execute.ALL, blocking=True, materialize_futures=True): |