Simplify VMobject.set_rgba_array

KristelCocoli · Feb 3, 2023 · 3d0fe27 · 3d0fe27
1 parent 4629e08
commit 3d0fe27
Showing 1 changed file with 4 additions and 12 deletions.
diff --git a/manimlib/mobject/types/vectorized_mobject.py b/manimlib/mobject/types/vectorized_mobject.py
@@ -166,20 +166,12 @@ def init_colors(self):
     def set_rgba_array(
         self,
         rgba_array: Vect4Array,
-        name: str | None = None,
+        name: str = "stroke_rgba",
         recurse: bool = False
     ) -> Self:
-        if name is None:
-            names = ["fill_rgba", "stroke_rgba"]
-        else:
-            names = [name]
-
-        for name in names:
-            super().set_rgba_array(rgba_array, name, recurse)
-            if name == "fill_rgba":
-                self.note_changed_fill()
-            elif name == "stroke_rgba":
-                self.note_changed_stroke()
+        super().set_rgba_array(rgba_array, name, recurse)
+        self.note_changed_fill()
+        self.note_changed_stroke()
         return self
 
     def set_fill(