read_precomputed: deal with potentially malformed info vertex attributes

navis-org · Oct 24, 2023 · 4225c0d · 4225c0d
1 parent 8ec6be7
commit 4225c0d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/navis/io/precomputed_io.py b/navis/io/precomputed_io.py
@@ -151,6 +151,10 @@ def read_buffer(
 
         swc = self.make_swc(nodes, edges)
 
+        # Check for malformed vertex attributes (should be list of dicts)
+        if isinstance(self.info.get('vertex_attributes', None), dict):
+            self.info['vertex_attributes'] = [self.info['vertex_attributes']]
+
         # Parse additional vertex attributes if specified as per the info file
         for attr in self.info.get('vertex_attributes', []):
             dtype = np.dtype(attr['data_type'])