File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
3231def 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
You can’t perform that action at this time.
0 commit comments