9b7f4d373f553711babfd13f6c67b3ddfe546a97,python/magent/model.py,NDArrayPackage,recv_from,#NDArrayPackage#Any#,79

Before Change

            conn.send_bytes(x)

    def recv_from(self, conn):
        bufs = [np.empty(shape=x[0], dtype=x[1]) for x in self.info]
        for buf in bufs:
            conn.recv_bytes_into(buf)
        return bufs

After Change

        for info in self.info:
            buf = np.empty(shape=(int(np.prod(info[0])),), dtype=info[1])

            item_size = int(np.prod(info[0][1:]))
            if np.prod(info[0]) > self.max_len:
                seg = int(self.max_len // item_size)
                for pt in range(0, int(np.prod(info[0])), seg * item_size):
                    conn.recv_bytes_into(buf[pt:pt+seg * item_size])
            else:
               conn.recv_bytes_into(buf)
            bufs.append(buf.reshape(info[0]))
        return bufs

Instances