changes to seed for tests
This commit is contained in:
parent
57274b8331
commit
e04dfb37fd
|
@ -10,6 +10,11 @@ from pytorch_lightning.testing import LightningTestModel
|
|||
from pytorch_lightning.logging import LightningLoggerBase, rank_zero_only
|
||||
from .test_models import get_hparams, get_test_tube_logger, init_save_dir, clear_save_dir
|
||||
|
||||
|
||||
RANDOM_FILE_PATHS = list(np.random.randint(12000, 19000, 1000))
|
||||
ROOT_SEED = 1234
|
||||
torch.manual_seed(ROOT_SEED)
|
||||
np.random.seed(ROOT_SEED)
|
||||
RANDOM_SEEDS = list(np.random.randint(0, 10000, 1000))
|
||||
|
||||
|
||||
|
@ -101,7 +106,7 @@ def test_mlflow_logger():
|
|||
|
||||
assert result == 1, "Training failed"
|
||||
|
||||
n = np.random.randint(0, 10000000, 1)[0]
|
||||
n = RANDOM_FILE_PATHS.pop()
|
||||
shutil.move(mlflow_dir, mlflow_dir + f'_{n}')
|
||||
|
||||
|
||||
|
@ -136,7 +141,7 @@ def test_mlflow_pickle():
|
|||
trainer2 = pickle.loads(pkl_bytes)
|
||||
trainer2.logger.log_metrics({"acc": 1.0})
|
||||
|
||||
n = np.random.randint(0, 10000000, 1)[0]
|
||||
n = RANDOM_FILE_PATHS.pop()
|
||||
shutil.move(mlflow_dir, mlflow_dir + f'_{n}')
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue