fix numpy seed key ignorance (#20)

ee13d84f · Ziyao Li · GitHub · 3aa106dc · ee13d84f
Unverified Commit ee13d84f authored Mar 05, 2023 by Ziyao Li Committed by GitHub Mar 05, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

unicore/data/data_utils.py unicore/data/data_utils.py +2 -2

No files found.
--- a/unicore/data/data_utils.py
+++ b/unicore/data/data_utils.py
@@ -94,8 +94,8 @@ def numpy_seed(seed, *addl_seeds, key=None):
        for s in addl_seeds:
            check_seed(s)
        seed = int(hash((seed, *addl_seeds)) % 1e8)
-        if key is not None:
-            seed = int(hash((seed, str_hash(key))) % 1e8)
+    if key is not None:
+        seed = int(hash((seed, str_hash(key))) % 1e8)
    state = np.random.get_state()
    np.random.seed(seed)
    try: