    def context(self, key):
        """
        Use this like:

        with timer.context("key"):
            # do stuff

        Then timer.tock("key") will be called automatically.
        """
        return _TimerContextManager(self, key)
Use this like:

with timer.context("key"):
# do stuff

Then timer.tock("key") will be called automatically.
serl_launcher/serl_launcher/utils/timer_utils.py