def sample(self):
action_pb = self.stub.Sample(gym_uds_pb2.Empty())
return action_pb.value
def reset(self):
state_pb = self.stub.Reset(gym_uds_pb2.Empty())
observation = np.asarray(state_pb.observation.data).reshape(
state_pb.observation.shape)
return observation