Method guard_node

workflow/executor/resource_manager.py:36–40

Acquire all resources required by the given node.

(self, node: Node)

    @contextmanager
    def guard_node(self, node: Node):
        """Acquire all resources required by the given node."""
        requests = self._resolve_node_requests(node)
        with self._acquire_resources(requests):
            yield

    def _resolve_node_requests(self, node: Node) -> List[ResourceRequest]:
        registration = get_node_registration(node.node_type)