From 15d1f0d7b1edc63edb06250a7d82f5dbbd69496b Mon Sep 17 00:00:00 2001 From: sneakers-the-rat Date: Wed, 4 Oct 2023 22:32:53 -0700 Subject: [PATCH] clear repo imports before returning from sys.module --- nwb_linkml/src/nwb_linkml/providers/schema.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/nwb_linkml/src/nwb_linkml/providers/schema.py b/nwb_linkml/src/nwb_linkml/providers/schema.py index 3daf228..04472f6 100644 --- a/nwb_linkml/src/nwb_linkml/providers/schema.py +++ b/nwb_linkml/src/nwb_linkml/providers/schema.py @@ -730,6 +730,10 @@ class PydanticProvider(Provider): module_name = self.module_name(namespace, version) namespace_name = module_name + '.namespace' + + if not allow_repo: + self._clear_package_imports() + if namespace_name in sys.modules: return sys.modules[namespace_name] @@ -741,8 +745,7 @@ class PydanticProvider(Provider): if path is None or not path.exists(): _ = self.build(namespace, version=version) - if not allow_repo: - self._clear_package_imports() + module = self.import_module(namespace, version) return module