diff --git a/ontobio/assoc_factory.py b/ontobio/assoc_factory.py index 0fb911aa..140b7c81 100644 --- a/ontobio/assoc_factory.py +++ b/ontobio/assoc_factory.py @@ -30,7 +30,7 @@ def __init__(self): initializes based on an ontology name """ - def create(self, ontology=None,subject_category=None,object_category=None,evidence=None,taxon=None,relation=None, file=None, fmt=None, skim=True): + def create(self, ontology=None,subject_category=None,object_category=None,evidence=None,taxon=None, relation=None, file=None, fmt=None, skim=True): """ creates an AssociationSet @@ -60,6 +60,7 @@ def create(self, ontology=None,subject_category=None,object_category=None,eviden logging.info("Fetching assocs from store") assocs = bulk_fetch_cached(subject_category=subject_category, object_category=object_category, + relation=relation, evidence=evidence, taxon=taxon) @@ -70,11 +71,12 @@ def create(self, ontology=None,subject_category=None,object_category=None,eviden amap = {} subject_label_map = {} for a in assocs: - rel = a['relation'] subj = a['subject'] subject_label_map[subj] = a['subject_label'] - amap[subj] = a['objects'] - + if subj in amap: + amap[subj].extend(a['objects']) + else: + amap[subj] = a['objects'] aset = AssociationSet(ontology=ontology, meta=meta,