Skip to content

Commit 34bbdf3

Browse files
committed
[Rec oracle] Use MT19937 generator
1 parent 7189d81 commit 34bbdf3

File tree

1 file changed

+6
-9
lines changed

1 file changed

+6
-9
lines changed

packages/examples/cvat/recording-oracle/src/handlers/process_intermediate_results.py

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -485,16 +485,13 @@ def process_intermediate_results( # noqa: PLR0912
485485
available_validation_frames = list(
486486
set(non_disabled_frames) - set(validation_frames_to_replace)
487487
)
488-
rng = np.random.default_rng()
488+
rng = np.random.Generator(np.random.MT19937())
489489
try:
490-
new_validation_frames = [
491-
int(f)
492-
for f in rng.choice(
493-
available_validation_frames,
494-
replace=False,
495-
size=len(validation_frames_to_replace),
496-
)
497-
]
490+
new_validation_frames = rng.choice(
491+
available_validation_frames,
492+
replace=False,
493+
size=len(validation_frames_to_replace),
494+
).tolist()
498495

499496
for prev_val_frame, new_val_frame in zip(
500497
validation_frames_to_replace, new_validation_frames, strict=True

0 commit comments

Comments
 (0)