Skip to content

Commit 1f2ce00

Browse files
pin random seed to resolve stochastic behavior
Signed-off-by: Brian Dellabetta <bdellabe@redhat.com>
1 parent 2d561d7 commit 1f2ce00

File tree

3 files changed

+8
-2
lines changed

3 files changed

+8
-2
lines changed

tests/lmeval/configs/vl_fp8_dynamic_per_token.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,4 +12,4 @@ lmeval:
1212
num_fewshot: 0
1313
batch_size: 8
1414
metrics:
15-
acc,none: 0.266
15+
acc,none: 0.333

tests/lmeval/configs/vl_w4a16_actorder_weight.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,5 +14,5 @@ lmeval:
1414
task: mmmu_val_economics
1515
num_fewshot: 0
1616
metrics:
17-
acc,none: 0.433
17+
acc,none: 0.366
1818
batch_size: 4

tests/lmeval/test_lmeval.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,11 @@
11
import os
2+
import random
23
import shutil
34
from pathlib import Path
45

56
import numpy
67
import pytest
8+
import torch
79
import yaml
810
from loguru import logger
911
from pydantic import BaseModel
@@ -57,6 +59,10 @@ class TestLMEval:
5759
""" # noqa: E501
5860

5961
def set_up(self):
62+
random.seed(42)
63+
numpy.random.seed(42)
64+
torch.manual_seed(42)
65+
6066
eval_config = yaml.safe_load(Path(TEST_DATA_FILE).read_text(encoding="utf-8"))
6167

6268
if os.environ.get("CADENCE", "commit") != eval_config.get("cadence"):

0 commit comments

Comments
 (0)