Ejemplo n.º 1
0
 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
Ejemplo n.º 2
0
 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