Commit 01e53935 by Haozheng Fan Committed by Tianqi Chen

hide psutil (#4013)

parent d7998d39
......@@ -39,7 +39,6 @@ import subprocess
import time
import sys
import signal
import psutil
from .._ffi.function import register_func
from .._ffi.base import py_str
......@@ -211,6 +210,7 @@ def _listen_loop(sock, port, rpc_key, tracker_addr, load_library, custom_addr):
server_proc.join(opts.get("timeout", None))
if server_proc.is_alive():
logger.info("Timeout in RPC session, kill..")
import psutil
parent = psutil.Process(server_proc.pid)
# terminate worker childs
for child in parent.children(recursive=True):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment