Utility for random string.

runhouse/utils.py

@@ -14,6 +14,7 @@
 import sys
 import tempfile
 import threading
+import uuid
 from concurrent.futures import ThreadPoolExecutor
 from datetime import datetime
 from enum import Enum
@@ -34,6 +35,13 @@
 logger = get_logger(__name__)
 
 
+def get_random_str(length: int = 8):
+    if length > 32:
+        raise ValueError("Max length of random string is 32")
+
+    return str(uuid.uuid4())[:length]
+
+
 ####################################################################################################
 # Simple env setup utilities
 ####################################################################################################

tests/fixtures/resource_fixtures.py

@@ -1,10 +1,10 @@
-import uuid
 from datetime import datetime
 
 import pytest
 
 from runhouse.globals import rns_client
 from runhouse.resources.resource import Resource
+from runhouse.utils import get_random_str
 from tests.conftest import init_args
 
 from tests.constants import TEST_ORG
@@ -38,7 +38,7 @@ def saved_resource_pool():
 
 
 def create_folder_path():
-    hash = uuid.uuid4().hex[:8]
+    hash = get_random_str(length=8)
     return f"test-{hash}-{datetime.now().strftime('%Y%m%d-%H%M%S')}"
 
 

tests/utils.py

@@ -2,7 +2,6 @@
 import importlib
 import json
 import os
-import uuid
 from datetime import datetime
 from pathlib import Path
 from typing import Dict, Optional
@@ -52,13 +51,6 @@ def get_pid_and_ray_node(a=0):
     return pid, node_id
 
 
-def get_random_str(length: int = 8):
-    if length > 32:
-        raise ValueError("Max length of random string is 32")
-
-    return str(uuid.uuid4())[:length]
-
-
 @contextlib.contextmanager
 def friend_account():
     """Used for the purposes of testing resource sharing among different accounts.