8c17debc9dc6552da8c317c0c852b55f03c80c61,slm_lab/spec/spec_util.py,,resolve_aeb,#Any#,141

Before Change

    body_num = _.get(spec, "body.num")

    if ae_product == "outer":
        aeb_coor_list = list(itertools.product(
            range(agent_num), range(env_num), range(body_num)))
    elif ae_product == "inner":
        ae_coor_itr = zip(range(agent_num), range(env_num))
        aeb_coor_list = list(itertools.product(
            ae_coor_itr, range(body_num)))

After Change

            aeb_list.extend(sub_aeb_list)
    else:  // custom AEB, body_num is a aeb_list
        aeb_list = [tuple(aeb) for aeb in body_num]
    aeb_list.sort()
    assert is_aeb_compact(
        aeb_list), "Failed check: for a, e, uniq count == len (shape), and for each a,e hash, b uniq count == b len (shape)"
    return aeb_list

Instances