Merge pull request #66 from CAVEconnectome/fix-colors-with-non

Handle null data with color selection
CAVEconnectome · Jan 30, 2025 · e74c2b0 · e74c2b0
2 parents 17977f0 + c8df47b
commit e74c2b0
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/nglui/statebuilder/mappers.py b/src/nglui/statebuilder/mappers.py
@@ -148,11 +148,12 @@ def seg_colors(self, data):
             for ii, oid in enumerate(self.fixed_ids):
                 colors[oid] = self.fixed_id_colors[ii]
 
-        if self.color_column is not None:
-            clist = data[self.color_column].to_list()
-            for col in self.data_columns:
-                for ii, oid in enumerate(data[col]):
-                    colors[oid] = clist[ii]
+        if data is not None:
+            if self.color_column is not None:
+                clist = data[self.color_column].to_list()
+                for col in self.data_columns:
+                    for ii, oid in enumerate(data[col]):
+                        colors[oid] = clist[ii]
 
         return colors