mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-05-21 20:51:01 -05:00
d8e8df81af
Fixed broken imports in system and module commands after removing dead command files: 1. System Command (system/system.py): - Removed imports: check, version, clean_workspace, report, protect - Kept: info, health, jupyter - Added 'doctor' as alias for comprehensive health check - Simplified to 4 subcommands: info, health, doctor, jupyter 2. Module Workflow Command (module/workflow.py): - Removed imports: view, test - Replaced ViewCommand._open_jupyter() with direct Jupyter Lab launch - Kept all module workflow functionality intact All 15 registered commands now load and execute successfully: ✅ Student: module, milestones, community, benchmark, olympics ✅ Developer: dev, system, src, package, nbgrader ✅ Shortcuts: export, test, grade, logo ✅ Essential: setup 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
95 lines
3.2 KiB
Python
95 lines
3.2 KiB
Python
"""
|
|
System command group for TinyTorch CLI: environment, configuration, and system tools.
|
|
"""
|
|
|
|
from argparse import ArgumentParser, Namespace
|
|
from rich.panel import Panel
|
|
|
|
from ..base import BaseCommand
|
|
from .info import InfoCommand
|
|
from .health import HealthCommand
|
|
from .jupyter import JupyterCommand
|
|
|
|
class SystemCommand(BaseCommand):
|
|
@property
|
|
def name(self) -> str:
|
|
return "system"
|
|
|
|
@property
|
|
def description(self) -> str:
|
|
return "System environment and configuration commands"
|
|
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
|
subparsers = parser.add_subparsers(
|
|
dest='system_command',
|
|
help='System subcommands',
|
|
metavar='SUBCOMMAND'
|
|
)
|
|
|
|
# Info subcommand
|
|
info_parser = subparsers.add_parser(
|
|
'info',
|
|
help='Show system and environment information'
|
|
)
|
|
info_cmd = InfoCommand(self.config)
|
|
info_cmd.add_arguments(info_parser)
|
|
|
|
# Health subcommand (quick check)
|
|
health_parser = subparsers.add_parser(
|
|
'health',
|
|
help='Quick environment health check'
|
|
)
|
|
health_cmd = HealthCommand(self.config)
|
|
health_cmd.add_arguments(health_parser)
|
|
|
|
# Doctor subcommand (alias for health - comprehensive validation)
|
|
doctor_parser = subparsers.add_parser(
|
|
'doctor',
|
|
help='Comprehensive environment validation and diagnosis'
|
|
)
|
|
doctor_cmd = HealthCommand(self.config)
|
|
doctor_cmd.add_arguments(doctor_parser)
|
|
|
|
# Jupyter subcommand
|
|
jupyter_parser = subparsers.add_parser(
|
|
'jupyter',
|
|
help='Start Jupyter notebook server'
|
|
)
|
|
jupyter_cmd = JupyterCommand(self.config)
|
|
jupyter_cmd.add_arguments(jupyter_parser)
|
|
|
|
def run(self, args: Namespace) -> int:
|
|
console = self.console
|
|
|
|
if not hasattr(args, 'system_command') or not args.system_command:
|
|
console.print(Panel(
|
|
"[bold cyan]System Commands[/bold cyan]\n\n"
|
|
"Available subcommands:\n"
|
|
" • [bold]info[/bold] - Show system/environment information\n"
|
|
" • [bold]health[/bold] - Quick environment health check\n"
|
|
" • [bold]doctor[/bold] - Comprehensive environment validation\n"
|
|
" • [bold]jupyter[/bold] - Start Jupyter notebook server\n\n"
|
|
"[dim]Example: tito system doctor[/dim]",
|
|
title="System Command Group",
|
|
border_style="bright_cyan"
|
|
))
|
|
return 0
|
|
|
|
# Execute the appropriate subcommand
|
|
if args.system_command == 'info':
|
|
cmd = InfoCommand(self.config)
|
|
return cmd.execute(args)
|
|
elif args.system_command in ('health', 'doctor'):
|
|
cmd = HealthCommand(self.config)
|
|
return cmd.execute(args)
|
|
elif args.system_command == 'jupyter':
|
|
cmd = JupyterCommand(self.config)
|
|
return cmd.execute(args)
|
|
else:
|
|
console.print(Panel(
|
|
f"[red]Unknown system subcommand: {args.system_command}[/red]",
|
|
title="Error",
|
|
border_style="red"
|
|
))
|
|
return 1
|