Skip to content

Commit 0e8c146

Browse files
authored
Revert D108449705
Differential Revision: D108905366 Pull Request resolved: pytorch#20340
1 parent a7953c5 commit 0e8c146

1 file changed

Lines changed: 2 additions & 7 deletions

File tree

backends/arm/test/conftest.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,6 @@ def pytest_configure(config):
2626
pytest._test_options["llama_inputs"] = config.option.llama_inputs # type: ignore[attr-defined]
2727

2828
logging.basicConfig(stream=sys.stdout)
29-
_set_random_seed()
3029

3130

3231
def pytest_collection_modifyitems(config, items):
@@ -79,28 +78,24 @@ def set_random_seed():
7978
ARM_TEST_SEED=3478246 pytest --config-file=/dev/null --verbose -s --color=yes backends/arm/test/ops/test_avg_pool.py -k <TESTCASE>
8079
8180
"""
82-
_set_random_seed()
83-
84-
85-
def _set_random_seed():
8681
import torch
8782

8883
seed_env = os.environ.get("ARM_TEST_SEED", "0")
8984
if seed_env == "RANDOM":
9085
random.seed() # reset seed, in case any other test has fiddled with it
9186
seed = random.randint(0, 2**32 - 1) # nosec B311 - non-crypto seed for tests
9287
torch.manual_seed(seed)
93-
print(f" ARM_TEST_SEED=RANDOM using:{seed} ", end=" ")
9488
elif str.isdigit(seed_env):
9589
seed = int(seed_env)
9690
random.seed(seed)
9791
torch.manual_seed(seed)
98-
print(f" ARM_TEST_SEED={seed} ", end=" ")
9992
else:
10093
raise TypeError(
10194
"ARM_TEST_SEED env variable must be integers or the string RANDOM"
10295
)
10396

97+
print(f" ARM_TEST_SEED={seed} ", end=" ")
98+
10499

105100
# ==== End of Pytest fixtures =====
106101

0 commit comments

Comments
 (0)