[vis] Fix stream plot component order

tum-pbs · Nov 13, 2023 · fedd5da · fedd5da
1 parent 6673bc4
commit fedd5da
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
@@ -439,7 +439,7 @@ def plot(self, data: Field, figure, subplot, space: Box, min_val: float, max_val
         x, y = reshaped_numpy(data.points, [vector, *data.shape.without('vector')])
         x = x[:, 0]
         y = y[0, :]
-        u, v = reshaped_numpy(data.values, [vector, *data.shape.without('vector')])
+        u, v = reshaped_numpy(data.values.vector[vector.item_names[0]], [vector, *data.shape.without('vector')])
         if (color == None).all:
             col = reshaped_numpy(math.vec_length(data.values), [*data.shape.without('vector')]).T
         else: