Improve stub generator

gentlegiantJGC · gentlegiantJGC · commit b82461d7c7e6 · 2025-09-18T08:56:14.000+01:00
diff --git a/tools/generate_pybind_stubs.py b/tools/generate_pybind_stubs.py
@@ -58,18 +58,18 @@ def eq_sub_func(match: re.Match) -> str:
         else:
             return "\n".join(
                 [
-                    f"{match.group('indent')}def __eq__(self, arg0: {match.group('other')}) -> {match.group('return')}:{match.group('ellipsis_docstring')}",
+                    f"{match.group('indent')}def __eq__(self, other: {match.group('other')}) -> {match.group('return')}:{match.group('ellipsis_docstring')}",
                     f"{match.group('indent')}@typing.overload",
-                    f"{match.group('indent')}def __eq__(self, arg0: typing.Any) -> bool | types.NotImplementedType: ...",
+                    f"{match.group('indent')}def __eq__(self, other: typing.Any) -> bool | types.NotImplementedType: ...",
                 ]
             )
     else:
         return "\n".join(
             [
                 f"{match.group('indent')}@typing.overload",
-                f"{match.group('indent')}def __eq__(self, arg0: {match.group('other')}) -> {match.group('return')}:{match.group('ellipsis_docstring')}",
+                f"{match.group('indent')}def __eq__(self, other: {match.group('other')}) -> {match.group('return')}:{match.group('ellipsis_docstring')}",
                 f"{match.group('indent')}@typing.overload",
-                f"{match.group('indent')}def __eq__(self, arg0: typing.Any) -> bool | types.NotImplementedType: ...",
+                f"{match.group('indent')}def __eq__(self, other: typing.Any) -> bool | types.NotImplementedType: ...",
             ]
         )
 

Original file line number	Diff line number	Diff line change
`@@ -58,18 +58,18 @@ def eq_sub_func(match: re.Match) -> str:`
`58`	`58`	`else:`
`59`	`59`	`return "\n".join(`
`60`	`60`	`[`
`61`		`- f"{match.group('indent')}def __eq__(self, arg0: {match.group('other')}) -> {match.group('return')}:{match.group('ellipsis_docstring')}",`
	`61`	`+ f"{match.group('indent')}def __eq__(self, other: {match.group('other')}) -> {match.group('return')}:{match.group('ellipsis_docstring')}",`
`62`	`62`	`f"{match.group('indent')}@typing.overload",`
`63`		`- f"{match.group('indent')}def __eq__(self, arg0: typing.Any) -> bool \| types.NotImplementedType: ...",`
	`63`	`+ f"{match.group('indent')}def __eq__(self, other: typing.Any) -> bool \| types.NotImplementedType: ...",`
`64`	`64`	`]`
`65`	`65`	`)`
`66`	`66`	`else:`
`67`	`67`	`return "\n".join(`
`68`	`68`	`[`
`69`	`69`	`f"{match.group('indent')}@typing.overload",`
`70`		`- f"{match.group('indent')}def __eq__(self, arg0: {match.group('other')}) -> {match.group('return')}:{match.group('ellipsis_docstring')}",`
	`70`	`+ f"{match.group('indent')}def __eq__(self, other: {match.group('other')}) -> {match.group('return')}:{match.group('ellipsis_docstring')}",`
`71`	`71`	`f"{match.group('indent')}@typing.overload",`
`72`		`- f"{match.group('indent')}def __eq__(self, arg0: typing.Any) -> bool \| types.NotImplementedType: ...",`
	`72`	`+ f"{match.group('indent')}def __eq__(self, other: typing.Any) -> bool \| types.NotImplementedType: ...",`
`73`	`73`	`]`
`74`	`74`	`)`
`75`	`75`