ray

Source code: tianshou/env/worker/ray.py

---

class RayEnvWorker(env_fn: Callable[[], Env | PettingZooEnv])

Ray worker used in RayVectorEnv.

close_env() → None

get_env_attr(key: str) → Any

recv() → tuple[ndarray, ndarray, ndarray, ndarray, ndarray]

Receive result from low-level worker.

If the last “send” function sends a NULL action, it only returns a single observation; otherwise it returns a tuple of (obs, rew, done, info) or (obs, rew, terminated, truncated, info), based on whether the environment is using the old step API or the new one.

render(**kwargs: Any) → Any

Render the environment.

reset(**kwargs: Any) → Any

seed(seed: int | None = None) → list[int] | None

send(action: ndarray | None, **kwargs: Any) → None

Send action signal to low-level worker.

When action is None, it indicates sending “reset” signal; otherwise it indicates “step” signal. The paired return value from “recv” function is determined by such kind of different signal.

set_env_attr(key: str, value: Any) → None

static wait(workers: list[RayEnvWorker], wait_num: int, timeout: float | None = None) → list[RayEnvWorker]

Given a list of workers, return those ready ones.