| 289 | ) |
| 290 | |
| 291 | class FuncWrapper: |
| 292 | def remote(self, *args, **kwargs): |
| 293 | return func_cls._remote( |
| 294 | args=args, |
| 295 | kwargs=kwargs, |
| 296 | serialized_runtime_env_info=serialized_runtime_env_info, |
| 297 | **updated_options, |
| 298 | ) |
| 299 | |
| 300 | @DeveloperAPI |
| 301 | def bind(self, *args, **kwargs): |
| 302 | """ |
| 303 | For Ray DAG building that creates static graph from decorated |
| 304 | class or functions. |
| 305 | """ |
| 306 | from ray.dag.function_node import FunctionNode |
| 307 | |
| 308 | return FunctionNode(func_cls._function, args, kwargs, updated_options) |
| 309 | |
| 310 | return FuncWrapper() |
| 311 |
no outgoing calls
no test coverage detected
searching dependent graphs…