Fixed typing annotations for dr.freeze

Author: DoeringChristian

drjit/__init__.py

@@ -1879,10 +1879,6 @@ def binary_search(start, end, pred):
 
     return start
 
-# Represents the type of the input arguments to the function
-P = ParamSpec("P")
-# Represents the Return value of the function
-R = TypeVar("R")
 # Represents the type of the state that is returned by ``state``
 S = TypeVar("S")
 # Represents the frozen function passed to the decorator without arguments
@@ -1891,9 +1887,7 @@ def binary_search(start, end, pred):
 F2 = TypeVar("F2")
 
 @overload
-def freeze(
-    f: None = None, *, state: Callable[P, S]
-) -> Callable[[Callable[P, R]], Callable[P, R]]:
+def freeze(f: None = None, *, state: Callable[..., S]) -> Callable[[F], F]:
     """
     Decorator to freeze a function for replaying kernels without compilation.