Skip to content

Commit 09fc551

Browse files
authored
Remove usage of sklearn from variants_generation.py (#223)
1 parent 4ffdd85 commit 09fc551

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

scripts/variants_generation.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
import csv
22
import numpy as np
3-
import sklearn.utils as skl_utils
43
from xlsxwriter.workbook import Workbook
54

65
NUM_TASKS = 3
@@ -17,7 +16,8 @@
1716
for j in range(int(NUM_STUDENTS / NUM_VARIANTS) + 1):
1817
list_of_variants.append(np.arange(NUM_VARIANTS) + 1)
1918
for variant in list_of_variants:
20-
shuffled_list_of_variants.append(skl_utils.shuffle(variant))
19+
np.random.shuffle(variant)
20+
shuffled_list_of_variants.append(variant)
2121
result_variants = np.concatenate(shuffled_list_of_variants)
2222
list_of_tasks.append(result_variants[:NUM_STUDENTS])
2323
str_of_print += '%d,'

0 commit comments

Comments
 (0)