Use relative import when importing self module

jameskermode · Jan 20, 2025 · cb83562 · cb83562
1 parent 10bc76e
commit cb83562
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/f90wrap/pywrapgen.py b/f90wrap/pywrapgen.py
@@ -721,7 +721,11 @@ def visit_Type(self, node):
             if node.parent.mod_name != node.mod_name:
                 cls_parent = "%s.%s" % (node.parent.mod_name, cls_parent)
                 if self.make_package:
-                    self.imports.add((self.py_mod_name, node.parent.mod_name))
+                    if self.relative:
+                        py_mod_name = '.'
+                    else:
+                        py_mod_name = self.py_mod_name
+                    self.imports.add((py_mod_name, node.parent.mod_name))
         self.write(
             '@f90wrap.runtime.register_class("%s.%s")' % (self.py_mod_name, cls_name)
         )