From 6a516737b55ce1e279b857751319450716b64d5d Mon Sep 17 00:00:00 2001 From: caufieldjh Date: Sun, 11 Aug 2024 13:58:24 -0400 Subject: [PATCH] Fix typing issues --- src/ontogpt/engines/knowledge_engine.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/ontogpt/engines/knowledge_engine.py b/src/ontogpt/engines/knowledge_engine.py index c8d482db7..ae40c7537 100644 --- a/src/ontogpt/engines/knowledge_engine.py +++ b/src/ontogpt/engines/knowledge_engine.py @@ -106,7 +106,7 @@ class KnowledgeEngine(ABC): mappers: Optional[List[BasicOntologyInterface]] = None """List of concept mappers, to assist in grounding to desired ID prefix""" - map_cache: Dict[str, str] = field(default_factory=dict) + map_cache: Dict[str, List[str]] = field(default_factory=dict) """Cache of mappings obtained using the specified mappers.""" labelers: Optional[List[BasicOntologyInterface]] = None @@ -425,7 +425,7 @@ def map_identifier(self, input_id: str, cls: ClassDefinition) -> Iterator[str]: for mapper in self.mappers: if isinstance(mapper, MappingProviderInterface): for mapping in mapper.sssom_mappings([input_id]): - self.map_cache[input_id].append(str(mapping.object_id)) + (self.map_cache[input_id]).append(str(mapping.object_id)) yield str(mapping.object_id) else: raise ValueError(f"Unknown mapper type {mapper}")