From looking at the source code, there is an even easier way to get the node restriction for the current node: ray.state.current_node_id()
Sadly nothing of this is to be found anywhere in the documentaiton. The API documentation seems to be restricted to a subset of modules and classes so even though this is in the PythonDoc of the respective function, it shows up nowhere in the API docs on the website.