map-predict_get_successors.py

def get_successors(h0: int, h1: int) -> Counter:
    cnt = get_cache_successors(h0, h1)
    if cnt is None:
        db = TrajDb()
        sql = "select t2 from triple where t0=? and t1=?"
        successors = [r[0] for r in db.query(sql, [int(h0), int(h1)])]
        cnt = Counter(successors)

        set_cache_successors(h0, h1, cnt)
    return cnt