graphic collections return Graphic instead of CollectionIndexer if sliced with int

kushalkolar · kushalkolar · commit 9409eb01c4ec · 2023-04-19T02:30:44.000-04:00
diff --git a/fastplotlib/graphics/_base.py b/fastplotlib/graphics/_base.py
@@ -356,8 +356,9 @@ def _reset_index(self):
             graphic.collection_index = new_index
 
     def __getitem__(self, key):
-        if isinstance(key, int):
-            key = [key]
+        if isinstance(key, (int, np.integer)):
+            # single graphic indexed
+            return self.graphics[key]
 
         if isinstance(key, slice):
             key = cleanup_slice(key, upper_bound=len(self))
