INNER CODE UNIT · Python

RPCServer

onnxsim/onnxsim · onnxsim/rpc/server.py:458

class RPCServer(socketserver.ThreadingTCPServer):
    allow_reuse_address = True
    daemon_threads = True

    def __init__(
        self,
        host: str = "127.0.0.1",
        port: int = 0,
        key: str = "",
        work_dir: Optional[str] = None,
        max_blob_bytes: int = proto.DEFAULT_MAX_BLOB_BYTES,
        verbose: bool = False,
    ):
        super().__init__((host, port), _Handler)
        self.key = key
        self.work_dir = work_dir or os.path.join(os.getcwd(), "onnxsim_rpc_workspace")
        os.makedirs(self.work_dir, exist_ok=True)
        self.max_blob_bytes = max_blob_bytes

View source record →