Skip to content

Commit 7a81ee8

Browse files
committed
set seed without transformers
1 parent 329cd26 commit 7a81ee8

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/autointent/context/data_handler/_safe_multilabel_stratification.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from __future__ import annotations
22

3+
import random
34
from typing import TYPE_CHECKING, Any
45

56
import numpy as np
@@ -113,9 +114,8 @@ def _iterative_stratify_remaining(
113114
return
114115
if random_seed is not None:
115116
# Workaround for buggy nature of IterativeStratification from skmultilearn
116-
from transformers import set_seed
117-
118-
set_seed(random_seed)
117+
random.seed(random_seed)
118+
np.random.seed(random_seed)
119119
splitter = IterativeStratification(
120120
n_splits=2,
121121
order=2,

0 commit comments

Comments
 (0)