Use set_color instead of set_rgba_array in vector_field

Liu-Yinlong · Feb 3, 2023 · 88959df · 88959df
1 parent 3d0fe27
commit 88959df
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/manimlib/mobject/vector_field.py b/manimlib/mobject/vector_field.py
@@ -13,6 +13,7 @@
 from manimlib.utils.bezier import interpolate
 from manimlib.utils.bezier import inverse_interpolate
 from manimlib.utils.color import get_colormap_list
+from manimlib.utils.color import rgb_to_color
 from manimlib.utils.dict_ops import merge_dicts_recursively
 from manimlib.utils.rate_functions import linear
 from manimlib.utils.simple_functions import sigmoid
@@ -173,7 +174,10 @@ def get_vector(self, coords: Iterable[float], **kwargs) -> Arrow:
             **vector_config
         )
         vect.shift(_input - origin)
-        vect.set_rgba_array([[*self.value_to_rgb(norm), self.opacity]])
+        vect.set_color(
+            rgb_to_color(self.value_to_rgb(norm)),
+            opacity=self.opacity,
+        )
         return vect