1818from ray_release .buildkite .output import buildkite_group , buildkite_open_last
1919from ray_release .cloud_util import archive_directory
2020from ray_release .cluster_manager .cluster_manager import ClusterManager
21- from ray_release .cluster_manager .full import FullClusterManager
2221from ray_release .cluster_manager .minimal import MinimalClusterManager
2322from ray_release .command_runner .anyscale_job_runner import AnyscaleJobRunner
2423from ray_release .command_runner .command_runner import CommandRunner
25- from ray_release .command_runner .job_runner import JobRunner
2624from ray_release .config import (
2725 DEFAULT_AUTOSUSPEND_MINS ,
2826 DEFAULT_BUILD_TIMEOUT ,
4846from ray_release .reporter .reporter import Reporter
4947from ray_release .result import Result , ResultStatus , handle_exception
5048from ray_release .signal_handling import (
51- register_handler ,
5249 reset_signal_handling ,
5350 setup_signal_handling ,
5451)
5552from ray_release .template import get_working_dir , load_test_cluster_compute
5653from ray_release .test import Test
5754
5855type_str_to_command_runner = {
59- "job" : JobRunner ,
6056 "anyscale_job" : AnyscaleJobRunner ,
6157}
6258
6359command_runner_to_cluster_manager = {
64- JobRunner : FullClusterManager ,
6560 AnyscaleJobRunner : MinimalClusterManager ,
6661}
6762
@@ -238,12 +233,6 @@ def _local_environment_information(
238233 cluster_id : Optional [str ],
239234 cluster_env_id : Optional [str ],
240235) -> None :
241- if isinstance (cluster_manager , FullClusterManager ):
242- if not no_terminate :
243- register_handler (
244- lambda sig , frame : cluster_manager .terminate_cluster (wait = True )
245- )
246-
247236 # Start cluster
248237 if cluster_id :
249238 buildkite_group (":rocket: Using existing cluster" )
@@ -258,10 +247,7 @@ def _local_environment_information(
258247
259248 cluster_manager .build_configs (timeout = build_timeout )
260249
261- if isinstance (cluster_manager , FullClusterManager ):
262- buildkite_group (":rocket: Starting up cluster" )
263- cluster_manager .start_cluster (timeout = cluster_timeout )
264- elif isinstance (command_runner , AnyscaleJobRunner ):
250+ if isinstance (command_runner , AnyscaleJobRunner ):
265251 command_runner .job_manager .cluster_startup_timeout = cluster_timeout
266252
267253 result .cluster_url = cluster_manager .get_cluster_url ()
0 commit comments