Upload random_utils.py with huggingface_hub

random_utils.py

@@ -5,35 +5,46 @@ import threading
 
 __default_seed__ = 42
 _thread_local = threading.local()
-_thread_local.seed = __default_seed__
-_thread_local.random = python_random.Random()
-random = _thread_local.random
 
 
-def set_seed(seed):
-    _thread_local.random.seed(seed)
-    _thread_local.seed = seed
+def get_seed():
+    try:
+        return _thread_local.seed
+    except AttributeError:
+        _thread_local.seed = __default_seed__
+        return _thread_local.seed
 
 
-def get_seed():
-    return _thread_local.seed
+def get_random():
+    try:
+        return _thread_local.random
+    except AttributeError:
+        _thread_local.random = python_random.Random(get_seed())
+        return _thread_local.random
+
+
+random = get_random()
+
+
+def set_seed(seed):
+    _thread_local.seed = seed
+    get_random().seed(seed)
 
 
 def get_random_string(length):
     letters = string.ascii_letters
-    result_str = "".join(random.choice(letters) for _ in range(length))
-    return result_str
+    return "".join(get_random().choice(letters) for _ in range(length))
 
 
 @contextlib.contextmanager
 def nested_seed(sub_seed=None):
-    state = _thread_local.random.getstate()
+    old_state = get_random().getstate()
     old_global_seed = get_seed()
     sub_seed = sub_seed or get_random_string(10)
     new_global_seed = str(old_global_seed) + "/" + sub_seed
     set_seed(new_global_seed)
     try:
-        yield _thread_local.random
+        yield get_random()
     finally:
         set_seed(old_global_seed)
-        _thread_local.random.setstate(state)
+        get_random().setstate(old_state)