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
