update_ids: deal with pandas StringArray

navis-org · Nov 14, 2023 · f366b92 · f366b92
1 parent 5b5830e
commit f366b92
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/fafbseg/flywire/segmentation.py b/fafbseg/flywire/segmentation.py
@@ -1059,6 +1059,8 @@ def update_ids(id,
                              '`root` column.')
     elif isinstance(id, pd.Series):
         id = id.values
+    elif isinstance(id, pd.core.arrays.string_.StringArray):
+        id = np.asarray(id)
 
     if isinstance(timestamp, str) and timestamp.startswith('mat'):
         client = get_cave_client(dataset=dataset)