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