From 206af21e908e5ea5105ad48ca63f07666c38253b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andreas=20Kl=C3=B6ckner?= Date: Thu, 22 May 2025 13:57:53 -0500 Subject: [PATCH] Fix stub generation for imported methods --- src/stubgen.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/stubgen.py b/src/stubgen.py index 5bb9bf3b..7c08d2d2 100755 --- a/src/stubgen.py +++ b/src/stubgen.py @@ -388,7 +388,12 @@ def put_function(self, fn: Callable[..., Any], name: Optional[str] = None, paren # Check if this function is an alias from *another* module if name and fn_module and fn_module != self.module.__name__: - self.put_value(fn, name) + if parent is None: + self.put_value(fn, name) + else: + import_name = self.import_object(fn_module, fn.__name__) + self.write_ln(f"{name} = {import_name}\n") + return # Check if this function is an alias from the *same* module