def to_json()

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

• 16 lines of code
• 1 McCabe index (conditional complexity)

    def to_json(self) -> Dict:
        """Convert the stats to a JSON-serializable dictionary."""
        return {
            "prompt_tokens": self.num_prompt_tokens,
            "generated_tokens": self.num_generated_tokens,
            "model_load_start_ms": self.model_load_start_ms,
            "model_load_end_ms": self.model_load_end_ms,
            "inference_start_ms": self.inference_start_ms,
            "token_encode_end_ms": self.token_encode_end_ms,
            "model_execution_start_ms": self.model_execution_start_ms,
            "model_execution_end_ms": self.model_execution_end_ms,
            "inference_end_ms": self.inference_end_ms,
            "prompt_eval_end_ms": self.prompt_eval_end_ms,
            "first_token_ms": self.first_token_ms,
            "aggregate_sampling_time_ms": self.aggregate_sampling_time_ms,
            "SCALING_FACTOR_UNITS_PER_SECOND": self.SCALING_FACTOR_UNITS_PER_SECOND,
        }