add take

interfaces/cap_minor.py

@@ -11,6 +11,11 @@ from huggingface_hub import HfApi
 from label_dicts import CAP_MIN_NUM_DICT, CAP_MIN_LABEL_NAMES, CAP_LABEL_NAMES 
 
 from .utils import is_disk_full
+from itertools import islice
+
+def take(n, iterable):
+    """Return the first n items of the iterable as a list."""
+    return list(islice(iterable, n))
 
 HF_TOKEN = os.environ["hf_read"]