fixed typing in utils

AIRI-Institute · Dec 20, 2023 · 1b61086 · 1b61086
1 parent 67db46d
commit 1b61086
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/probing/ud_filter/utils.py b/probing/ud_filter/utils.py
@@ -151,7 +151,9 @@ def determine_ud_savepath(
     return Path(final_path)
 
 
-def delete_duplicates(probing_dict: Dict[str, List[str]]) -> Dict[str, List[str]]:
+def delete_duplicates(
+    probing_dict: Dict[str, List[Tuple[str, List[int]]]]
+) -> Dict[str, List[Tuple[str, List[int]]]]:
     """Deletes sentences with more than one different classes of node_pattern found"""
 
     all_sent = [sent for cl_sent in probing_dict.values() for sent, inds in cl_sent]