Merge pull request #432 from dkuettel/typing

dont destroy type of decorated function

Author: Erotemic

line_profiler/autoprofile/profmod_extractor.py

@@ -157,7 +157,8 @@ def _ast_get_imports_from_tree(
                     tree_index (int):
                         the index of the import as found in the tree
         """
-        module_dict_list = []
+        module_dict_list: list[dict[str, str | int | None]] = []
+        module_dict: dict[str, str | int | None]
         modname_list = []
         for idx, node in enumerate(tree.body):
             if isinstance(node, ast.Import):

line_profiler/explicit_profiler.py

@@ -171,7 +171,7 @@ def func4():
 import pathlib
 import sys
 import typing
-from typing import Any, Callable, TypeVar
+from typing import Any, Callable, ParamSpec, TypeVar
 
 if typing.TYPE_CHECKING:
     ConfigArg = str | pathlib.PurePath | bool | None
@@ -183,6 +183,8 @@ def func4():
 from .toml_config import ConfigSource
 
 F = TypeVar('F', bound=Callable[..., Any])
+P = ParamSpec('P')
+R = TypeVar('R')
 
 # The first process that enables profiling records its PID here. Child processes
 # created via multiprocessing (spawn/forkserver) inherit this environment value,
@@ -427,7 +429,7 @@ def disable(self) -> None:
         """
         self.enabled = False
 
-    def __call__(self, func: Callable) -> Callable:
+    def __call__(self, func: Callable[P, R]) -> Callable[P, R]:
         """
         If the global profiler is enabled, decorate a function to start the
         profiler on function entry and stop it on function exit. Otherwise