from functools import lru_cache from pathlib import Path @lru_cache() def _is_ray_cluster(): """Checks if the bootstrap config file exists. This will always exist if using an autoscaling cluster/started with the ray cluster launcher. """ return Path("~/ray_bootstrap_config.yaml").expanduser().exists()