@@ -54,19 +54,19 @@ def _get_job_storage_root(package_path: str) -> str:
5454 return job_storage_root
5555
5656
57- def _get_grpc_host_and_port (package_path : str ):
57+ def _get_grpc_host_and_port (package_path : str ) -> ( str , int ) :
5858 fed_config = _get_server_fed_config (package_path )
5959 server_conf = fed_config ["servers" ][0 ]
6060 grpc_service_config = server_conf ["service" ]
6161 grpc_target_address = grpc_service_config ["target" ]
62- host , port = grpc_target_address .split (":" )
63- return host , int (port )
62+ _ , port = grpc_target_address .split (":" )
63+ return "localhost" , int (port )
6464
6565
66- def _get_admin_host_and_port (package_path : str ):
66+ def _get_admin_host_and_port (package_path : str ) -> ( str , int ) :
6767 fed_config = _get_server_fed_config (package_path )
6868 server_conf = fed_config ["servers" ][0 ]
69- return server_conf [ "admin_host" ] , int (server_conf ["admin_port" ])
69+ return "localhost" , int (server_conf ["admin_port" ])
7070
7171
7272class ServerPackageChecker (PackageChecker ):
@@ -103,8 +103,8 @@ def get_dry_run_command(self) -> str:
103103 self .job_storage_root = _get_job_storage_root (self .package_path )
104104 return command
105105
106- def stop_dry_run (self ):
107- super ().stop_dry_run ()
106+ def stop_dry_run (self , force = True ):
107+ super ().stop_dry_run (force = force )
108108 if os .path .exists (self .snapshot_storage_root ):
109109 shutil .rmtree (self .snapshot_storage_root )
110110 if os .path .exists (self .job_storage_root ):
0 commit comments