We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 329cd26 commit 7a81ee8Copy full SHA for 7a81ee8
src/autointent/context/data_handler/_safe_multilabel_stratification.py
@@ -1,5 +1,6 @@
1
from __future__ import annotations
2
3
+import random
4
from typing import TYPE_CHECKING, Any
5
6
import numpy as np
@@ -113,9 +114,8 @@ def _iterative_stratify_remaining(
113
114
return
115
if random_seed is not None:
116
# Workaround for buggy nature of IterativeStratification from skmultilearn
- from transformers import set_seed
117
-
118
- set_seed(random_seed)
+ random.seed(random_seed)
+ np.random.seed(random_seed)
119
splitter = IterativeStratification(
120
n_splits=2,
121
order=2,
0 commit comments