Fix typing

Author: lantiga

.actions/assistant.py

@@ -486,12 +486,13 @@ def convert_version2nightly(ver_file: str = "src/version.info") -> None:
     import sys
 
     import jsonargparse
+    from jsonargparse import ArgumentParser
 
-    def _parse_known_args_patch(self, args: Any = None, namespace: Any = None) -> tuple[Any]:
-        namespace, args = super(jsonargparse.ArgumentParser, self)._parse_known_args(args, namespace, intermixed=False)
+    def _parse_known_args_patch(self: ArgumentParser, args: Any = None, namespace: Any = None) -> tuple[Any, Any]:
+        namespace, args = super(ArgumentParser, self)._parse_known_args(args, namespace, intermixed=False)
         return namespace, args
 
     if sys.version_info >= (3, 12, 8):
-        setattr(jsonargparse.ArgumentParser, "_parse_known_args", _parse_known_args_patch)
+        setattr(ArgumentParser, "_parse_known_args", _parse_known_args_patch)
 
     jsonargparse.CLI(AssistantCLI, as_positional=False)

src/lightning/pytorch/cli.py

@@ -48,7 +48,7 @@
         set_config_read_mode,
     )
 
-    def _parse_known_args_patch(self, args: Any = None, namespace: Any = None) -> tuple[Any]:
+    def _parse_known_args_patch(self: ArgumentParser, args: Any = None, namespace: Any = None) -> tuple[Any, Any]:
         namespace, args = super(ArgumentParser, self)._parse_known_args(args, namespace, intermixed=False)
         return namespace, args