[vis] Fix Matplotlib VectorCloud2D plot

tum-pbs · Feb 6, 2025 · 2148b2e · 2148b2e
1 parent 56caba7
commit 2148b2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/phi/vis/_matplotlib/_matplotlib_plots.py b/phi/vis/_matplotlib/_matplotlib_plots.py
@@ -455,7 +455,7 @@ def plot(self, data: Field, figure, subplot, space: Box, min_val: float, max_val
             x, y = reshaped_numpy(c_data.center[dims], [vector, c_data.shape.without('vector')])
             u, v = reshaped_numpy(c_data.values.vector[dims], [vector, c_data.shape.without('vector')])
             color_i = color[idx]
-            if (color[idx] == 'cmap').all:
+            if wrap(color[idx] == 'cmap').all:
                 col = _next_line_color(subplot, kind='collections')  # ToDo
             elif color[idx].shape:
                 col = [_plt_col(c) for c in color_i.numpy(c_data.shape.non_channel).reshape(-1)]