ft_utils

Reviewed By: jermenkoo

Differential Revision: D63884573

fbshipit-source-id: dae670155fc426f7b088b09747d08f885dd1f717

Author: generatedunixname89002005287564

atomic_bench.py

@@ -11,7 +11,7 @@
 
 
 class LockedReference:
-    def __init__(self, value: Optional[Any]) -> None:  # pyre-ignore[2]
+    def __init__(self, value: Any | None) -> None:  # pyre-ignore[2]
         self._value = value
         self._lock = threading.Lock()
 

benchmark_utils.py

@@ -10,8 +10,9 @@
 import threading
 import time
 import traceback
+from collections.abc import Callable, Sequence
 
-from typing import Callable, List, Sequence, Type, TypeVar
+from typing import List, Type, TypeVar
 
 from ft_utils.local import BatchExecutor
 
@@ -90,17 +91,17 @@ def benchmark_operation(operation_func: Callable[[], None]) -> float:
 def worker(
     operation_func: Callable[[], None],
     barrier: threading.Barrier,
-) -> List[float]:
+) -> list[float]:
     """
     Executes the benchmark multiple times and collects run times.
     """
     barrier.wait()  # Synchronize the start of operations
-    run_times: List[float] = [benchmark_operation(operation_func) for _ in range(5)]
+    run_times: list[float] = [benchmark_operation(operation_func) for _ in range(5)]
     return run_times
 
 
 def execute_benchmarks(
-    provider_class: Type[BenchmarkProvider],
+    provider_class: type[BenchmarkProvider],
 ) -> None:
     """
     Sets up and executes benchmarks across multiple threads using methods from a BenchmarkProvider.

concurrent.py

@@ -3,7 +3,8 @@
 # pyre-strict
 
 import threading
-from typing import Any, Iterator, Optional
+from collections.abc import Iterator
+from typing import Any, Optional
 
 from ft_utils._concurrent import AtomicInt64, AtomicReference, ConcurrentDict
 
@@ -38,7 +39,7 @@ class ConcurrentGatheringIterator:
     scaling (Optional(int)): expected number of threads or cores accessing the structure.
     """
 
-    def __init__(self, scaling: Optional[int] = None) -> None:
+    def __init__(self, scaling: int | None = None) -> None:
         if scaling is not None:
             self._dict: ConcurrentDict = ConcurrentDict(scaling)
         else:
@@ -107,7 +108,7 @@ def iterator_local(self, max_key: int, clear: bool = True) -> Iterator[Any]:  #
 
 
 class ConcurrentQueue:
-    def __init__(self, scaling: Optional[int] = None) -> None:
+    def __init__(self, scaling: int | None = None) -> None:
         if scaling is not None:
             self._dict: ConcurrentDict = ConcurrentDict(scaling)
         else:

concurrent_dict_bench.py

@@ -15,8 +15,8 @@
 class ConcurretDictBenchmarkProvider(BenchmarkProvider):
     def __init__(self, operations: int) -> None:
         self._operations = operations
-        self._cdct: Optional[ConcurrentDict] = None
-        self._dct: Optional[dict[int | str, int]] = None
+        self._cdct: ConcurrentDict | None = None
+        self._dct: dict[int | str, int] | None = None
 
     def set_up(self) -> None:
         self._cdct = ConcurrentDict(os.cpu_count())

examples/consistent_counter.py

@@ -4,7 +4,7 @@
 
 
 import threading
-from typing import Callable
+from collections.abc import Callable
 
 from ft_utils.concurrent import AtomicInt64
 from ft_utils.synchronization import IntervalLock

examples/fibonacci.py

@@ -51,7 +51,7 @@ def compute():  # pyre-ignore
         f.result()
 
 
-def fib_worker(n: int, memo: Dict[int, Tuple[int, int]]) -> Tuple[int, int]:
+def fib_worker(n: int, memo: dict[int, tuple[int, int]]) -> tuple[int, int]:
     # Check memoization cache in a thread-safe manner
     if n in memo:
         cached.incr()

lock_bench.py

@@ -3,7 +3,7 @@
 # pyre-strict
 
 import threading
-from typing import Callable
+from collections.abc import Callable
 
 from ft_utils.benchmark_utils import BenchmarkProvider, execute_benchmarks
 from ft_utils.local import LocalWrapper

lock_test_utils.py

@@ -7,7 +7,7 @@
 import threading
 import time
 import unittest
-from typing import Callable
+from collections.abc import Callable
 
 from ft_utils.concurrent import AtomicFlag, AtomicReference
 

random_bench.py

@@ -4,7 +4,7 @@
 
 import random
 import threading
-from typing import Callable
+from collections.abc import Callable
 
 from ft_utils.benchmark_utils import BenchmarkProvider, execute_benchmarks
 from ft_utils.local import BatchExecutor, LocalWrapper

test_run_all.py

@@ -13,9 +13,7 @@ def run_test(filename):
     if f_tail != ".py":
         raise ValueError(f"filename `{filename}` is not a Python (.py) file")
     module = f"ft_utils.tests.{f_head}"
-    result = subprocess.run(
-        ["python", "-m", module], stdout=subprocess.PIPE, stderr=subprocess.PIPE
-    )
+    result = subprocess.run(["python", "-m", module], capture_output=True)
     if result.returncode != 0:
         print(f"{module} failed:")
         print(result.stdout.decode())

tsp_bench.py

@@ -5,8 +5,9 @@
 import sys
 import threading
 import time
+from collections.abc import Callable
 from concurrent.futures import Future, ThreadPoolExecutor
-from typing import Any, Callable
+from typing import Any
 
 # pyre-strict