def reset()

in optimum/executorch/stats.py [0:0]

• 15 lines of code
• 2 McCabe index (conditional complexity)

    def reset(self, all_stats: bool = False):
        """Reset stats, optionally including model load times."""
        if all_stats:
            self.model_load_start_ms = 0
            self.model_load_end_ms = 0

        self.inference_start_ms = 0
        self.token_encode_end_ms = 0
        self.model_execution_start_ms = 0
        self.model_execution_end_ms = 0
        self.prompt_eval_end_ms = 0
        self.first_token_ms = 0
        self.inference_end_ms = 0
        self.aggregate_sampling_time_ms = 0
        self.num_prompt_tokens = 0
        self.num_generated_tokens = 0
        self._aggregate_sampling_timer_start_timestamp = 0