Add fixture torch_seed to fixtures/torch.py

vocalpy · Feb 15, 2023 · 122e564 · 122e564
1 parent ff761ef
commit 122e564
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/fixtures/torch.py b/tests/fixtures/torch.py
@@ -24,3 +24,12 @@ def get_test_dtypes() -> Dict[str, torch.dtype]:
 @pytest.fixture()
 def dtype(dtype_name) -> torch.dtype:
     return TEST_DTYPES[dtype_name]
+
+
+TORCH_SEED = 42
+
+
+@pytest.fixture()
+def torch_seed():
+    """Value for seed used in tests"""
+    return TORCH_SEED