✨ Accept nanobind module for stubgen

mypy/stubdoc.py

@@ -163,13 +163,14 @@ def add_token(self, token: tokenize.TokenInfo) -> None:
                 self.arg_type = self.accumulator
                 self.state.pop()
             elif self.state[-1] == STATE_ARGUMENT_LIST:
-                self.arg_name = self.accumulator
-                if not (
-                    token.string == ")" and self.accumulator.strip() == ""
-                ) and not _ARG_NAME_RE.match(self.arg_name):
-                    # Invalid argument name.
-                    self.reset()
-                    return
+                if not self.accumulator == "/":
+                    self.arg_name = self.accumulator
+                    if not (
+                        token.string == ")" and self.accumulator.strip() == ""
+                    ) and not _ARG_NAME_RE.match(self.arg_name):
+                        # Invalid argument name.
+                        self.reset()
+                        return
 
             if token.string == ")":
                 self.state.pop()

mypy/stubgenc.py

@@ -38,6 +38,7 @@
     "Optional",
     "Tuple",
     "Union",
+    "Sequence",
 )
 
 
@@ -194,7 +195,7 @@ def generate_stub_for_c_module(
     done = set()
     items = sorted(get_members(module), key=lambda x: x[0])
     for name, obj in items:
-        if is_c_function(obj):
+        if is_c_function(obj) or is_nanobind_function(obj):
             generate_c_function_stub(
                 module,
                 name,
@@ -307,6 +308,14 @@ def is_c_type(obj: object) -> bool:
     return inspect.isclass(obj) or type(obj) is type(int)
 
 
+def is_nanobind_function(obj: object) -> bool:
+    return (
+        hasattr(type(obj), "__module__")
+        and type(obj).__module__ == "nanobind"
+        and type(obj).__name__ == "nb_func"
+    )
+
+
 def is_pybind11_overloaded_function_docstring(docstr: str, name: str) -> bool:
     return docstr.startswith(f"{name}(*args, **kwargs)\n" + "Overloaded function.\n\n")