From c6fc24cda5e563f77884377d0274818ab450a819 Mon Sep 17 00:00:00 2001 From: caufieldjh Date: Wed, 10 Jan 2024 17:24:28 -0500 Subject: [PATCH] Update init of SPIRESEngine for engine resolver --- src/ontogpt/engines/resolver.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/ontogpt/engines/resolver.py b/src/ontogpt/engines/resolver.py index 79476b916..a529c437e 100644 --- a/src/ontogpt/engines/resolver.py +++ b/src/ontogpt/engines/resolver.py @@ -6,6 +6,7 @@ from ontogpt.engines.halo_engine import HALOEngine # type: ignore from ontogpt.engines.knowledge_engine import KnowledgeEngine from ontogpt.engines.spires_engine import SPIRESEngine +from ontogpt.io.template_loader import get_template_details resolver = ClassResolver([SPIRESEngine, HALOEngine], base=KnowledgeEngine) @@ -15,7 +16,7 @@ def create_engine( ) -> Union[KnowledgeEngine, SPIRESEngine]: """Create a knowledge engine.""" if engine is None: - engine = SPIRESEngine + engine = SPIRESEngine(template_details=get_template_details(template=template)) if isinstance(engine, str): engine = resolver.get_class(engine)(**kwargs) if engine is not None and not isinstance(engine, str):