chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,46 @@
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
"""``lmcache tool cache-simulator simulate`` subcommand."""
|
||||
|
||||
# Standard
|
||||
import argparse
|
||||
|
||||
# First Party
|
||||
from lmcache.cli.commands.base import BaseCommand
|
||||
|
||||
|
||||
class SimulateCommand(BaseCommand):
|
||||
"""
|
||||
Replay logs at a fixed cache capacity; print a text report and s
|
||||
ave a 7-panel statistics PNG.
|
||||
"""
|
||||
|
||||
def name(self) -> str:
|
||||
return "simulate"
|
||||
|
||||
def help(self) -> str:
|
||||
return (
|
||||
"Replay logs at a fixed cache capacity; print a text report "
|
||||
"and save a 7-panel statistics PNG."
|
||||
)
|
||||
|
||||
def add_arguments(self, parser: argparse.ArgumentParser) -> None:
|
||||
# First Party
|
||||
from lmcache.tools.cache_simulator.simulator import add_simulate_arguments
|
||||
|
||||
add_simulate_arguments(parser)
|
||||
|
||||
def register(self, subparsers: argparse._SubParsersAction) -> None:
|
||||
# Skip _add_output_args — this command defines its own --output.
|
||||
# Gracefully skip if optional dependencies are missing.
|
||||
try:
|
||||
parser = subparsers.add_parser(self.name(), help=self.help())
|
||||
self.add_arguments(parser)
|
||||
parser.set_defaults(func=self.execute)
|
||||
except ImportError:
|
||||
return
|
||||
|
||||
def execute(self, args: argparse.Namespace) -> None:
|
||||
# First Party
|
||||
from lmcache.tools.cache_simulator.simulator import run_simulate
|
||||
|
||||
run_simulate(args)
|
||||
Reference in New Issue
Block a user