formatting py3.8

rizavelioglu · Oct 9, 2023 · 0f80892 · 0f80892
1 parent b4403b2
commit 0f80892
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/fashionfail/models/prediction_utils.py b/src/fashionfail/models/prediction_utils.py
@@ -1,6 +1,6 @@
 import json
 from pathlib import Path
-from typing import Any, Dict, List
+from typing import Any, Dict, List, Tuple
 
 import numpy as np
 import pandas as pd
@@ -77,7 +77,7 @@ def load_tpu_preds(path_to_preds: str, preprocess: bool = True) -> pd.DataFrame:
 
 def _filter_preds_for_classes(
     row: pd.Series, class_ids: List[int]
-) -> tuple[np.ndarray, np.ndarray, np.ndarray, List[Dict[str, Any]]]:
+) -> Tuple[np.ndarray, np.ndarray, np.ndarray, List[Dict[str, Any]]]:
     """
     Filter prediction attributes based on class IDs.