GPJ: fixing test_utils.py

aidotse · Dec 29, 2024 · f2104e3 · f2104e3
1 parent a378f84
commit f2104e3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...tests_synthetic_data_attacks/tests_syn_text_pii_scanner/data_tests/aux_data_test_utils.py b/...tests_synthetic_data_attacks/tests_syn_text_pii_scanner/data_tests/aux_data_test_utils.py
@@ -1,6 +1,6 @@
 """Auxiliary data for test_utils.py suite of tests."""
-from typing import Any, Dict, List
 import os
+from typing import Any, Dict, List
 
 from numpy import array
 from transformers import PreTrainedTokenizerFast

diff --git a/leakpro/tests/tests_synthetic_data_attacks/tests_syn_text_pii_scanner/test_utils.py b/leakpro/tests/tests_synthetic_data_attacks/tests_syn_text_pii_scanner/test_utils.py
@@ -269,5 +269,5 @@ def test_compare_piis_lists(*, verbose: bool) -> None:
     e_items = aux.e_items_test_compare_piis_lists_fact()
     assert sorted_sim_items == e_items["e_sorted_sim_items"]
     assert distr.keys() == e_items["e_distr"].keys()
-    for k in distr.keys():
+    for k in distr:
         assert abs(distr[k] - e_items["e_distr"][k])<0.000001