reuse numpy for .pyf()

briangu · Dec 19, 2023 · 7fe50fd · 7fe50fd
1 parent 4bcf424
commit 7fe50fd
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 8 deletions.
diff --git a/examples/math/trig.kg b/examples/math/trig.kg
@@ -0,0 +1,3 @@
+.pyf("numpy";"cos")
+
+.p(cos(30))
diff --git a/klongpy/sys_fn.py b/klongpy/sys_fn.py
@@ -325,14 +325,17 @@ def _import_module(klong, x, from_list=None):
             print(f".py: {e}")
             raise RuntimeError(f"module could not be imported: {x}")
     else:
-        spec = importlib.util.find_spec(x)
-        if spec is not None:
-            try:
-                module = importlib.util.module_from_spec(spec)
-                spec.loader.exec_module(module)
-            except Exception as e:
-                print(f".py: {e}")
-                raise RuntimeError(f"module could not be imported: {x}")
+        if x in sys.modules:
+            module = sys.modules[x]
+        else:
+            spec = importlib.util.find_spec(x)
+            if spec is not None:
+                try:
+                    module = importlib.util.module_from_spec(spec)
+                    spec.loader.exec_module(module)
+                except Exception as e:
+                    print(f".py: {e}")
+                    raise RuntimeError(f"module could not be imported: {x}")
     try:
         klong_exports = module.__dict__.get("klong_exports")
         if klong_exports is None: