Upload collections.py with huggingface_hub

collections.py

@@ -4,7 +4,7 @@ from typing import Dict, List
 
 from .artifact import Artifact
 from .dataclass import AbstractField
-from .random_utils import random
+from .random_utils import get_random
 
 
 class Collection(Artifact):
@@ -13,8 +13,8 @@ class Collection(Artifact):
     def __getitem__(self, key):
         try:
             return self.items[key]
-        except LookupError:
-            raise LookupError(f"Cannot find item {repr(key)} in {repr(self)}")
+        except LookupError as e:
+            raise LookupError(f"Cannot find item {key!r} in {self!r}") from e
 
 
 class ListCollection(Collection):
@@ -43,13 +43,18 @@ class ItemPicker(Artifact):
     def __call__(self, collection: Collection):
         try:
             return collection[int(self.item)]
-        except (SyntaxError, KeyError, ValueError) as e:  # in case picking from a dictionary
+        except (
+            SyntaxError,
+            KeyError,
+            ValueError,
+        ):  # in case picking from a dictionary
             return collection[self.item]
 
 
 class RandomPicker(Artifact):
     def __call__(self, collection: Collection):
         if isinstance(collection, ListCollection):
-            return random.choice(list(collection.items))
-        elif isinstance(collection, DictCollection):
-            return random.choice(list(collection.items.values()))
+            return get_random().choice(list(collection.items))
+        if isinstance(collection, DictCollection):
+            return get_random().choice(list(collection.items.values()))
+        return None