Update util.py

AllenInstitute · Oct 30, 2024 · 82377b0 · 82377b0
1 parent e3e6f25
commit 82377b0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/deep_neurographs/utils/util.py b/src/deep_neurographs/utils/util.py
@@ -477,7 +477,10 @@ def sample_once(my_container):
     sample
 
     """
-    return sample(my_container, 1)[0]
+    if type(my_container) is list:
+        return sample(my_container, 1)[0]
+    else:
+        return sample(list(my_container), 1)[0]
 
 
 # --- dictionary utils ---