Import ABC from collections.abc for Python 3.10 compatibility. (#193)

nubank · Mar 22, 2022 · a5409ae · a5409ae
1 parent ff06c2b
commit a5409ae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/fklearn/metrics/pd_extractors.py b/src/fklearn/metrics/pd_extractors.py
@@ -1,4 +1,4 @@
-import collections
+import collections.abc
 from datetime import datetime
 from itertools import chain, repeat
 
@@ -74,7 +74,7 @@ def stability_curve_evaluator_extractor(result, base_extractor):
 
 @curry
 def repeat_split_log(split_log, results_len):
-    if isinstance(split_log, collections.Iterable):
+    if isinstance(split_log, collections.abc.Iterable):
         n_repeat = results_len // len(split_log)
         # The logic below makes [1, 2, 3] into [1, 1, 1, 2, 2, 2, 3, 3, 3] for n_repeat=3
         return list(chain.from_iterable(zip(*repeat(split_log, n_repeat))))