Update app.py

app.py

@@ -17,8 +17,9 @@
 __author__ = 'Dmitry Ustalov'
 __license__ = 'Apache 2.0'
 
+from collections.abc import Callable
 from functools import partial
-from typing import IO, Tuple, List, cast, Dict, Set, Callable
+from typing import BinaryIO, cast
 
 import gradio as gr
 import networkx as nx
@@ -71,13 +72,13 @@ def bradley_terry(wins: npt.NDArray[np.int_], ties: npt.NDArray[np.int_],
     return p
 
 
-def centrality(algorithm: Callable[[nx.DiGraph], Dict[int, float]],
+def centrality(algorithm: Callable[[nx.DiGraph], dict[int, float]],
                wins: npt.NDArray[np.int_], ties: npt.NDArray[np.int_]) -> npt.NDArray[np.float_]:
     A = wins + .5 * ties
 
     G = nx.from_numpy_array(A, create_using=nx.DiGraph)
 
-    scores: Dict[int, float] = algorithm(G)
+    scores: dict[int, float] = algorithm(G)
 
     p = np.array([scores[i] for i in range(len(G))])
 
@@ -162,15 +163,15 @@ ALGORITHMS = {
 }
 
 
-def largest_strongly_connected_component(df: pd.DataFrame) -> Set[str]:
+def largest_strongly_connected_component(df: pd.DataFrame) -> set[str]:
     G = nx.from_pandas_edgelist(df, source='left', target='right', create_using=nx.DiGraph)
     H = nx.from_pandas_edgelist(df[df['winner'] == 'tie'], source='right', target='left', create_using=nx.DiGraph)
     F = nx.compose(G, H)
     largest = max(nx.strongly_connected_components(F), key=len)
-    return cast(Set[str], largest)
+    return cast(set[str], largest)
 
 
-def handler(file: IO[bytes], algorithm: str, filtered: bool, truncated: bool, seed: int) -> Tuple[pd.DataFrame, Figure]:
+def handler(file: BinaryIO, algorithm: str, filtered: bool, truncated: bool, seed: int) -> tuple[pd.DataFrame, Figure]:
     if file is None:
         raise gr.Error('File must be uploaded')
 
@@ -258,7 +259,7 @@ def main() -> None:
                 label='Comparisons'
             ),
             gr.Dropdown(
-                choices=cast(List[str], ALGORITHMS),
+                choices=cast(list[str], ALGORITHMS),
                 value='Bradley-Terry (1952)',
                 label='Algorithm'
             ),