fix: fix colored static CSV import

skybrush-io · Jul 10, 2023 · 3d74ece · 3d74ece
1 parent 89c0061
commit 3d74ece
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/src/modules/sbstudio/plugin/operators/add_markers_from_static_csv.py b/src/modules/sbstudio/plugin/operators/add_markers_from_static_csv.py
@@ -83,21 +83,13 @@ def execute_on_formation(self, formation, context):
             )
             light_effect = light_effects.active_entry
             light_effect.output = "INDEXED_BY_FORMATION"
+            colors = [item.color.as_vector() for item in imported_data.values()]
             image = light_effect.create_color_image(
                 name="Image for light effect '{}'".format(formation.name),
                 width=1,
                 height=len(colors),
             )
-            image.pixels.foreach_set(
-                list(
-                    chain(
-                        *[
-                            list(item.color.as_vector())
-                            for item in imported_data.values()
-                        ]
-                    )
-                )
-            )
+            image.pixels.foreach_set(list(chain(*colors)))
 
         return {"FINISHED"}