Replace obsolete types (#644)

The tool `pyupgrade` was used to modernize the code base given that
frequenz-SDK is already using python 3.11 as minimum version. The tool
was mainly used to replace obsolete types.

Just as a reference the tool was run with the following shell command:

```sh
find ./src ./benchmarks ./docs ./examples ./tests -name '*.py' -exec pyupgrade --py311-plus {} \;
```

Also `black` and `isort` were both run to format the code base after
running `pyupgrade`.

Fixes #433

Author: daniel-zullo-frequenz

benchmarks/power_distribution/power_distributor.py

@@ -7,8 +7,9 @@
 import csv
 import random
 import timeit
+from collections.abc import Coroutine
 from datetime import timedelta
-from typing import Any, Coroutine, Dict, List, Set  # pylint: disable=unused-import
+from typing import Any
 
 from frequenz.channels import Broadcast
 
@@ -32,7 +33,7 @@
 PORT = 61060
 
 
-async def send_requests(batteries: Set[int], request_num: int) -> List[Result]:
+async def send_requests(batteries: set[int], request_num: int) -> list[Result]:
     """Send requests to the PowerDistributingActor and wait for the response.
 
     Args:
@@ -47,7 +48,7 @@ async def send_requests(batteries: Set[int], request_num: int) -> List[Result]:
     """
     battery_pool = microgrid.battery_pool(batteries)
     results_rx = battery_pool.power_distribution_results()
-    result: List[Result] = []
+    result: list[Result] = []
     for _ in range(request_num):
         await battery_pool.set_power(Power(float(random.randrange(100000, 1000000))))
         try:
@@ -61,7 +62,7 @@ async def send_requests(batteries: Set[int], request_num: int) -> List[Result]:
     return result
 
 
-def parse_result(result: List[List[Result]]) -> Dict[str, float]:
+def parse_result(result: list[list[Result]]) -> dict[str, float]:
     """Parse result.
 
     Args:
@@ -91,8 +92,8 @@ def parse_result(result: List[List[Result]]) -> Dict[str, float]:
 
 async def run_test(  # pylint: disable=too-many-locals
     num_requests: int,
-    batteries: Set[int],
-) -> Dict[str, Any]:
+    batteries: set[int],
+) -> dict[str, Any]:
     """Run test.
 
     Args:
@@ -112,7 +113,7 @@ async def run_test(  # pylint: disable=too-many-locals
         requests_receiver=power_request_channel.new_receiver(),
         battery_status_sender=battery_status_channel.new_sender(),
     ):
-        tasks: List[Coroutine[Any, Any, List[Result]]] = []
+        tasks: list[Coroutine[Any, Any, list[Result]]] = []
         tasks.append(send_requests(batteries, num_requests))
 
         result = await asyncio.gather(*tasks)
@@ -133,7 +134,7 @@ async def run() -> None:
         HOST, PORT, ResamplerConfig(resampling_period=timedelta(seconds=1.0))
     )
 
-    all_batteries: Set[Component] = connection_manager.get().component_graph.components(
+    all_batteries: set[Component] = connection_manager.get().component_graph.components(
         component_category={ComponentCategory.BATTERY}
     )
     batteries_ids = {c.component_id for c in all_batteries}

benchmarks/timeseries/benchmark_datasourcing.py

@@ -14,7 +14,7 @@
 import sys
 import tracemalloc
 from time import perf_counter
-from typing import Any, Tuple
+from typing import Any
 
 from frequenz.channels import Broadcast, Receiver, ReceiverStoppedError
 
@@ -131,7 +131,7 @@ async def consume(channel: Receiver[Any]) -> None:
         )
 
 
-def parse_args() -> Tuple[int, int, bool]:
+def parse_args() -> tuple[int, int, bool]:
     """Parse the command line arguments.
 
     Returns:

benchmarks/timeseries/benchmark_ringbuffer.py

@@ -6,7 +6,7 @@
 import random
 import timeit
 from datetime import datetime, timedelta, timezone
-from typing import Any, Dict, TypeVar
+from typing import Any, TypeVar
 
 import numpy as np
 
@@ -66,7 +66,7 @@ def test_slices(days: int, buffer: OrderedRingBuffer[Any], median: bool) -> None
                 total += float(np.average(minutes))
 
 
-def test_29_days_list(num_runs: int) -> Dict[str, float]:
+def test_29_days_list(num_runs: int) -> dict[str, float]:
     """Run the 29 day test on the list backend."""
     days = 29
     buffer = OrderedRingBuffer([0.0] * MINUTES_IN_29_DAYS, timedelta(minutes=1))
@@ -76,7 +76,7 @@ def test_29_days_list(num_runs: int) -> Dict[str, float]:
     return {"fill": fill_time, "test": test_time}
 
 
-def test_29_days_array(num_runs: int) -> Dict[str, float]:
+def test_29_days_array(num_runs: int) -> dict[str, float]:
     """Run the 29 day test on the array backend."""
     days = 29
     buffer = OrderedRingBuffer(
@@ -91,7 +91,7 @@ def test_29_days_array(num_runs: int) -> Dict[str, float]:
     return {"fill": fill_time, "test": test_time}
 
 
-def test_29_days_slicing_list(num_runs: int) -> Dict[str, float]:
+def test_29_days_slicing_list(num_runs: int) -> dict[str, float]:
     """Run slicing tests on list backend."""
     days = 29
     buffer = OrderedRingBuffer([0.0] * MINUTES_IN_29_DAYS, timedelta(minutes=1))
@@ -107,7 +107,7 @@ def test_29_days_slicing_list(num_runs: int) -> Dict[str, float]:
     return {"fill": fill_time, "median": median_test_time, "avg": avg_test_time}
 
 
-def test_29_days_slicing_array(num_runs: int) -> Dict[str, float]:
+def test_29_days_slicing_array(num_runs: int) -> dict[str, float]:
     """Run slicing tests on array backend."""
     days = 29
     buffer = OrderedRingBuffer(

benchmarks/timeseries/periodic_feature_extractor.py

@@ -9,7 +9,6 @@
 implementation.
 """
 
-from __future__ import annotations
 
 import asyncio
 import collections.abc
@@ -18,7 +17,6 @@
 from datetime import datetime, timedelta, timezone
 from functools import partial
 from timeit import timeit
-from typing import List
 
 import numpy as np
 from frequenz.channels import Broadcast
@@ -79,7 +77,7 @@ def _calculate_avg_window_py(
     feature_extractor: PeriodicFeatureExtractor,
     window: NDArray[np.float_],
     window_size: int,
-    weights: List[float] | None = None,
+    weights: list[float] | None = None,
 ) -> NDArray[np.float_]:
     """
     Plain python version of the average calculator.

benchmarks/timeseries/resampling.py

@@ -3,9 +3,9 @@
 
 """Benchmark resampling."""
 
+from collections.abc import Sequence
 from datetime import datetime, timedelta, timezone
 from timeit import timeit
-from typing import Sequence
 
 from frequenz.sdk.timeseries import Sample
 from frequenz.sdk.timeseries._quantities import Quantity

benchmarks/timeseries/ringbuffer_memusage.py

@@ -3,7 +3,6 @@
 
 """Memory allocation benchmark for the ringbuffer."""
 
-from __future__ import annotations
 
 import argparse
 import tracemalloc

benchmarks/timeseries/ringbuffer_serialization.py

@@ -3,7 +3,6 @@
 
 """Benchmarks the serialization of the `OrderedRingBuffer` class."""
 
-from __future__ import annotations
 
 import fnmatch
 import os

examples/battery_pool.py

@@ -3,12 +3,11 @@
 
 """Script with an example how to use BatteryPool."""
 
-from __future__ import annotations
 
 import asyncio
 import logging
 from datetime import timedelta
-from typing import Any, Dict
+from typing import Any
 
 from frequenz.channels import Receiver
 from frequenz.channels.util import MergeNamed
@@ -32,7 +31,7 @@ async def main() -> None:
     )
 
     battery_pool = microgrid.battery_pool()
-    receivers: Dict[str, Receiver[Any]] = {
+    receivers: dict[str, Receiver[Any]] = {
         "soc": battery_pool.soc.new_receiver(maxsize=1),
         "capacity": battery_pool.capacity.new_receiver(maxsize=1),
         "power_bounds": battery_pool.power_bounds.new_receiver(maxsize=1),

src/frequenz/sdk/_internal/_asyncio.py

@@ -3,7 +3,6 @@
 
 """General purpose async tools."""
 
-from __future__ import annotations
 
 import asyncio
 from abc import ABC

src/frequenz/sdk/_internal/_singleton_meta.py

@@ -4,13 +4,13 @@
 """Definition of Singleton metaclass."""
 
 from threading import Lock
-from typing import Any, Dict
+from typing import Any
 
 
 class SingletonMeta(type):
     """This is a thread-safe implementation of Singleton."""
 
-    _instances: Dict[Any, type] = {}
+    _instances: dict[Any, type] = {}
     """The dictionary of instances of the singleton classes."""
 
     _lock: Lock = Lock()