Method time

nemo_rl/utils/timer.py:110–123

Context manager for timing a block of code. Args: label: The label to use for this timing Yields: None

(self, label: str)

    @contextmanager
    def time(self, label: str) -> Generator[None, None, None]:
        """Context manager for timing a block of code.

        Args:
            label: The label to use for this timing

        Yields:
            None
        """
        self.start(label)
        try:
            yield
        finally:
            self.stop(label)

    def get_elapsed(self, label: str) -> list[float]:
        """Get all elapsed time measurements for a specific label.