feat: Add module load method to guppy

CQCL · Jan 19, 2024 · 95d11ba · 95d11ba
1 parent bd4ff8b
commit 95d11ba
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/guppylang/decorator.py b/guppylang/decorator.py
@@ -258,6 +258,13 @@ def dummy(*args: Any, **kwargs: Any) -> Any:
 
         return dec
 
+    def load(self, m: ModuleType | GuppyModule) -> None:
+        caller = self._get_python_caller()
+        if caller not in self._modules:
+            self._modules[caller] = GuppyModule(caller.name)
+        module = self._modules[caller]
+        module.load(m)
+
     def take_module(self, id: ModuleIdentifier | None = None) -> GuppyModule:
         """Returns the local GuppyModule, removing it from the local state."""
         orig_id = id