def _MakeMachine(machines): if isinstance(machines, str): machines = [machines] rp_machine = env_pb.EnvProto().machine for m_data in machines: m = rp_machine.add() if isinstance(m_data, str): m.addr = m_data elif isinstance(m_data, dict): if "addr" in m_data: m.addr = m_data["addr"] if "ctrl_port_agent" in m_data: m.ctrl_port_agent = m_data["ctrl_port_agent"] if "data_port_agent" in m_data: m.data_port_agent = m_data["data_port_agent"] else: raise NotImplementedError id = 0 addrs_for_check = set() for m in rp_machine: m.id = id id += 1 assert m.addr not in addrs_for_check addrs_for_check.add(m.addr) return rp_machine
def _DefaultEnvProto(): env_proto = env_pb.EnvProto() machine = env_proto.machine.add() machine.id = 0 machine.addr = "127.0.0.1" env_proto.grpc_use_no_signal = True return env_proto
def _DefaultEnvProto(): env_proto = env_pb.EnvProto() machine = env_proto.machine.add() machine.id = 0 machine.addr = "127.0.0.1" return env_proto