Skip to content

Commit

Permalink
Log execution time of scene rendering in the Manim Checkhealth command
Browse files Browse the repository at this point in the history
  • Loading branch information
chopan050 committed Jul 13, 2024
1 parent 628a545 commit 4afc856
Showing 1 changed file with 14 additions and 2 deletions.
16 changes: 14 additions & 2 deletions manim/cli/checkhealth/commands.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,12 @@
from __future__ import annotations

import sys
import time

import click
import cloup

from .checks import HEALTH_CHECKS
from manim.cli.checkhealth.checks import HEALTH_CHECKS

__all__ = ["checkhealth"]

Expand Down Expand Up @@ -62,7 +63,13 @@ def checkhealth():
import manim as mn

class CheckHealthDemo(mn.Scene):
def __init__(self):
super().__init__()
self.execution_time = None

def construct(self):
start = time.time()

banner = mn.ManimBanner().shift(mn.UP * 0.5)
self.play(banner.create())
self.wait(0.5)
Expand All @@ -79,5 +86,10 @@ def construct(self):
mn.FadeOut(text_tex_group, shift=mn.DOWN),
)

self.execution_time = time.time() - start

with mn.tempconfig({"preview": True, "disable_caching": True}):
CheckHealthDemo().render()
scene = CheckHealthDemo()
scene.render()

click.echo(f"Scene rendered in {scene.execution_time:.2f} seconds.")

0 comments on commit 4afc856

Please sign in to comment.