Use ParamSpec to improve typing for actors

Previously, the actor decorator was returning `typing.Any` objects.

This commit introduces a `ParamSpec` and a `TypeVar`, to improve
typing of actor params, and returned type.

Signed-off-by: Sahas Subramanian <[email protected]>

Author: shsms

src/frequenz/sdk/actor/_decorator.py

@@ -12,7 +12,9 @@
 import asyncio
 import inspect
 import logging
-from typing import Any, Optional, Type, TypeVar
+from typing import Any, Callable, Optional, Type, TypeVar
+
+from typing_extensions import ParamSpec
 
 from frequenz.sdk._internal._asyncio import cancel_and_await
 
@@ -55,7 +57,11 @@ class BaseActor:
     restart_limit: Optional[int] = None
 
 
-def actor(cls: Type[Any]) -> Type[Any]:
+_P = ParamSpec("_P")
+_R = TypeVar("_R")
+
+
+def actor(cls: Callable[_P, _R]) -> Type[_R]:
     """Decorate a class into a simple composable actor.
 
     A actor using the `actor` decorator should define an `async def run(self)`
@@ -181,7 +187,7 @@ async def run(self) -> None:
     class ActorClass(cls, BaseActor):  # type: ignore
         """A wrapper class to make an actor."""
 
-        def __init__(self, *args: Any, **kwargs: Any) -> None:
+        def __init__(self, *args: _P.args, **kwargs: _P.kwargs) -> None:
             """Create an `ActorClass` instance.
 
             Also call __init__ on `cls`.