From 3d0fe27c553f566fe58b5c9daf4e81483e01f9ab Mon Sep 17 00:00:00 2001 From: Grant Sanderson Date: Thu, 2 Feb 2023 18:23:41 -0800 Subject: [PATCH] Simplify VMobject.set_rgba_array --- manimlib/mobject/types/vectorized_mobject.py | 16 ++++------------ 1 file changed, 4 insertions(+), 12 deletions(-) diff --git a/manimlib/mobject/types/vectorized_mobject.py b/manimlib/mobject/types/vectorized_mobject.py index 8b203f0e..ce4eae34 100644 --- a/manimlib/mobject/types/vectorized_mobject.py +++ b/manimlib/mobject/types/vectorized_mobject.py @@ -166,20 +166,12 @@ class VMobject(Mobject): 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(