diff --git a/bionty/_bionty.py b/bionty/_bionty.py index ac0c7af..8c8a669 100644 --- a/bionty/_bionty.py +++ b/bionty/_bionty.py @@ -15,7 +15,7 @@ def create_or_get_organism_record( - organism: str | Record | None, registry: type[Record] + organism: str | Record | None, registry: type[Record], field: str | None = None ) -> Record | None: # return None if an Record doesn't have organism field organism_record = None @@ -49,6 +49,11 @@ def create_or_get_organism_record( organism_record = None if organism_record is None: + if hasattr(registry, "_ontology_id_field") and field in { + registry._ontology_id_field, + "uid", + }: + return None raise AssertionError( f"{registry.__name__} requires to specify a organism name via `organism=` or `bionty.settings.organism=`!" )