Upload collections.py with huggingface_hub

collections.py

@@ -1,8 +1,8 @@
-import random
 from abc import abstractmethod
 from dataclasses import field
 
 from .artifact import Artifact
+from .random_utils import random
 
 
 class Collection(Artifact):
@@ -20,6 +20,15 @@ class ListCollection(Collection):
     def __len__(self):
         return len(self.items)
 
+    def append(self, item):
+        self.items.append(item)
+
+    def extend(self, other):
+        self.items.extend(other.items)
+
+    def __add__(self, other):
+        return ListCollection(self.items.__add__(other.items))
+
 
 class DictCollection(Collection):
     items: dict = field(default_factory=dict)
@@ -32,7 +41,10 @@ class ItemPicker(Artifact):
     item: object = None
 
     def __call__(self, collection: Collection):
-        return collection[self.item]
+        try:
+            return collection[int(self.item)]
+        except (SyntaxError, KeyError, ValueError) as e:  # in case picking from a dictionary
+            return collection[self.item]
 
 
 class RandomPicker(Artifact):