def prepare_local_env(self, ray_wheels_url: Optional[str] = None):
# Install matching Ray for job submission
try:
install_matching_ray_locally(ray_wheels_url
or os.environ.get("RAY_WHEELS", None))
except Exception as e:
raise LocalEnvSetupError(
f"Error setting up local environment: {e}") from e
def prepare_local_env(self, ray_wheels_url: Optional[str] = None):
try:
install_matching_ray_locally(
ray_wheels_url or os.environ.get("RAY_WHEELS", None)
)
install_cluster_env_packages(self.cluster_manager.cluster_env)
except Exception as e:
raise LocalEnvSetupError(f"Error setting up local environment: {e}") from e