def _retrieve_ts_server_process()

in src/sagemaker_pytorch_serving_container/torchserve.py [0:0]

• 10 lines of code
• 5 McCabe index (conditional complexity)

def _retrieve_ts_server_process():
    ts_server_processes = list()

    for process in psutil.process_iter():
        if TS_NAMESPACE in process.cmdline():
            ts_server_processes.append(process)

    if not ts_server_processes:
        raise Exception("Torchserve model server was unsuccessfully started")

    if len(ts_server_processes) > 1:
        raise Exception("multiple ts model servers are not supported")

    return ts_server_processes[0]