Upgrade to modern Python syntax (#1956)

* Upgrade to modern Python syntax

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: cclauss

docs/source/manim_directive.py

@@ -202,7 +202,7 @@ def run(self):
             ".. code-block:: python",
             "",
             "    from manim import *\n",
-            *["    " + line for line in self.content],
+            *("    " + line for line in self.content),
         ]
         source_block = "\n".join(source_block)
 
@@ -286,7 +286,7 @@ def _write_rendering_stats(scene_name, run_time, file_name):
     with open(rendering_times_file_path, "a") as file:
         csv.writer(file).writerow(
             [
-                re.sub("^(reference\/)|(manim\.)", "", file_name),
+                re.sub(r"^(reference\/)|(manim\.)", "", file_name),
                 scene_name,
                 "%.3f" % run_time,
             ]
@@ -302,15 +302,15 @@ def _log_rendering_times(*args):
 
             print("\nRendering Summary\n-----------------\n")
 
-            max_file_length = max([len(row[0]) for row in data])
+            max_file_length = max(len(row[0]) for row in data)
             for key, group in it.groupby(data, key=lambda row: row[0]):
                 key = key.ljust(max_file_length + 1, ".")
                 group = list(group)
                 if len(group) == 1:
                     row = group[0]
                     print(f"{key}{row[2].rjust(7, '.')}s {row[1]}")
                     continue
-                time_sum = sum([float(row[2]) for row in group])
+                time_sum = sum(float(row[2]) for row in group)
                 print(
                     f"{key}{f'{time_sum:.3f}'.rjust(7, '.')}s  => {len(group)} EXAMPLES"
                 )

example_scenes/basic.py

@@ -29,7 +29,7 @@ def construct(self):
         transform_title.to_corner(UP + LEFT)
         self.play(
             Transform(title, transform_title),
-            LaggedStart(*[FadeOut(obj, shift=DOWN) for obj in basel]),
+            LaggedStart(*(FadeOut(obj, shift=DOWN) for obj in basel)),
         )
         self.wait()
 

manim/animation/animation.py

@@ -106,11 +106,9 @@ def __new__(
             if func is not None:
                 anim = func(mobject, *args, **kwargs)
                 logger.debug(
-                    (
-                        f"The {cls.__name__} animation has been is overridden for "
-                        f"{type(mobject).__name__} mobjects. use_override = False can "
-                        f" be used as keyword argument to prevent animation overriding."
-                    )
+                    f"The {cls.__name__} animation has been is overridden for "
+                    f"{type(mobject).__name__} mobjects. use_override = False can "
+                    f" be used as keyword argument to prevent animation overriding."
                 )
                 return anim
         return super().__new__(cls)
@@ -224,7 +222,7 @@ def get_all_mobjects(self) -> Sequence[Mobject]:
 
     def get_all_families_zipped(self) -> Iterable[Tuple]:
         return zip(
-            *[mob.family_members_with_points() for mob in self.get_all_mobjects()]
+            *(mob.family_members_with_points() for mob in self.get_all_mobjects())
         )
 
     def update_mobjects(self, dt: float) -> None:

manim/animation/creation.py

@@ -500,12 +500,12 @@ def __init__(
         self.time_per_char = time_per_char
         tpc = self.time_per_char
         anims = it.chain(
-            *[
+            *(
                 [
                     ShowIncreasingSubsets(word, run_time=tpc * len(word)),
                     Animation(word, run_time=0.005 * len(word) ** 1.5),
                 ]
                 for word in text_mobject
-            ]
+            )
         )
         super().__init__(*anims, **kwargs)

manim/animation/indication.py

@@ -327,7 +327,7 @@ def __init__(self, vmobject, n_segments=10, time_width=0.1, remover=True, **kwar
         max_time_width = kwargs.pop("time_width", self.time_width)
         AnimationGroup.__init__(
             self,
-            *[
+            *(
                 ShowPassingFlash(
                     vmobject.deepcopy().set_stroke(width=stroke_width),
                     time_width=time_width,
@@ -337,7 +337,7 @@ def __init__(self, vmobject, n_segments=10, time_width=0.1, remover=True, **kwar
                     np.linspace(0, max_stroke_width, self.n_segments),
                     np.linspace(max_time_width, 0, self.n_segments),
                 )
-            ],
+            ),
         )
 
 

manim/animation/transform.py

@@ -136,7 +136,7 @@ def get_all_families_zipped(self) -> Iterable[tuple]:  # more precise typing?
             self.starting_mobject,
             self.target_copy,
         ]
-        return zip(*[mob.family_members_with_points() for mob in mobs])
+        return zip(*(mob.family_members_with_points() for mob in mobs))
 
     def interpolate_submobject(
         self,

manim/camera/camera.py

@@ -678,7 +678,7 @@ def set_cairo_context_color(self, ctx, rgbas, vmobject):
         else:
             points = vmobject.get_gradient_start_and_end_points()
             points = self.transform_points_pre_display(vmobject, points)
-            pat = cairo.LinearGradient(*it.chain(*[point[:2] for point in points]))
+            pat = cairo.LinearGradient(*it.chain(*(point[:2] for point in points)))
             step = 1.0 / (len(rgbas) - 1)
             offsets = np.arange(0, 1 + step, step)
             for rgba, offset in zip(rgbas, offsets):