cf4379b2fc6ec480a801397aa8481f6a55ba5c6e,graphbrain/meaning/ontology.py,,subtypes,#,4

Before Change

def subtypes(hg, edge):
    ont_edges = hg.search((const.type_of_pred, "*", edge))
    return tuple([ont_edge[1] for ont_edge in ont_edges])


def supertypes(hg, edge):
    ont_edges = hg.serch((const.type_of_pred, edge, "*"))

After Change

def subtypes(hg, edge, deep=False):
    ont_edges = hg.search((const.type_of_pred, "*", edge))
    subs = set([ont_edge[1] for ont_edge in ont_edges])
    if deep:
        new_subs = set()
        for sub in subs:
            new_subs |= subtypes(hg, sub, deep=True)
        subs |= new_subs
    return subs


def supertypes(hg, edge, deep=False):
    ont_edges = hg.search((const.type_of_pred, edge, "*"))

Instances