Function get_cond_inds

sat/sample_video.py:175–179

(n_cond, n_round=0)

# - if n_round = 1, cond_inds = [-3, -2, -1]

def get_cond_inds(n_cond, n_round=0):
    if n_round == 0:
        return list(range(n_cond))
    else:
        return [-3, -2, -1]


def set_seed(seed: int):