mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-04-30 02:30:43 -05:00
e159a65118
- Fix AttributeError when calling reset without --all flag - Use getattr() with defaults for 'all' and 'list_backups' attributes
127 lines
4.4 KiB
Python
127 lines
4.4 KiB
Python
"""
|
|
Info command for TinyTorch CLI: shows system and environment information.
|
|
"""
|
|
|
|
from argparse import ArgumentParser, Namespace
|
|
import sys
|
|
import os
|
|
import platform
|
|
import shutil
|
|
from pathlib import Path
|
|
from rich.panel import Panel
|
|
from rich.table import Table
|
|
|
|
from .base import BaseCommand
|
|
|
|
class InfoCommand(BaseCommand):
|
|
@property
|
|
def name(self) -> str:
|
|
return "info"
|
|
|
|
@property
|
|
def description(self) -> str:
|
|
return "Show system and environment information"
|
|
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
|
# No arguments needed
|
|
pass
|
|
|
|
def run(self, args: Namespace) -> int:
|
|
console = self.console
|
|
|
|
console.print()
|
|
console.print(Panel(
|
|
"💻 TinyTorch System & Environment Information",
|
|
title="System Info",
|
|
border_style="bright_cyan"
|
|
))
|
|
console.print()
|
|
|
|
# System Information Table
|
|
info_table = Table(title="System Details", show_header=True, header_style="bold cyan")
|
|
info_table.add_column("Component", style="yellow", width=25)
|
|
info_table.add_column("Value", style="white", width=50)
|
|
|
|
# Python
|
|
python_version = f"{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}"
|
|
info_table.add_row("Python Version", python_version)
|
|
|
|
# Platform
|
|
system_name = platform.system()
|
|
system_release = platform.release()
|
|
machine = platform.machine()
|
|
info_table.add_row("Platform", f"{system_name} {system_release} ({machine})")
|
|
|
|
# Working Directory
|
|
working_dir = Path.cwd()
|
|
info_table.add_row("Working Directory", str(working_dir))
|
|
|
|
# Virtual Environment
|
|
venv_exists = self.venv_path.exists()
|
|
in_venv = (
|
|
os.environ.get('VIRTUAL_ENV') is not None or
|
|
(hasattr(sys, 'base_prefix') and sys.base_prefix != sys.prefix) or
|
|
hasattr(sys, 'real_prefix')
|
|
)
|
|
|
|
if venv_exists and in_venv:
|
|
venv_status = "✅ Active"
|
|
venv_path = os.environ.get('VIRTUAL_ENV', str(self.venv_path))
|
|
elif venv_exists:
|
|
venv_status = "⚠️ Exists (Not Active)"
|
|
venv_path = str(self.venv_path)
|
|
else:
|
|
venv_status = "❌ Not Found"
|
|
venv_path = "N/A"
|
|
|
|
info_table.add_row("Virtual Environment", venv_status)
|
|
if venv_path != "N/A":
|
|
info_table.add_row(" └─ Path", venv_path)
|
|
|
|
# TinyTorch Version
|
|
try:
|
|
import tinytorch
|
|
tinytorch_version = getattr(tinytorch, '__version__', 'unknown')
|
|
tinytorch_path = Path(tinytorch.__file__).parent
|
|
info_table.add_row("TinyTorch Version", tinytorch_version)
|
|
info_table.add_row(" └─ Location", str(tinytorch_path))
|
|
except ImportError:
|
|
info_table.add_row("TinyTorch Version", "❌ Not Installed")
|
|
|
|
# Disk Space
|
|
try:
|
|
disk_usage = shutil.disk_usage(working_dir)
|
|
free_gb = disk_usage.free / (1024**3)
|
|
total_gb = disk_usage.total / (1024**3)
|
|
used_percent = (disk_usage.used / disk_usage.total) * 100
|
|
info_table.add_row("Disk Space", f"{free_gb:.1f} GB free / {total_gb:.1f} GB total ({used_percent:.1f}% used)")
|
|
except Exception:
|
|
info_table.add_row("Disk Space", "Unable to determine")
|
|
|
|
# Memory
|
|
try:
|
|
import psutil
|
|
mem = psutil.virtual_memory()
|
|
free_gb = mem.available / (1024**3)
|
|
total_gb = mem.total / (1024**3)
|
|
used_percent = mem.percent
|
|
info_table.add_row("Memory", f"{free_gb:.1f} GB free / {total_gb:.1f} GB total ({used_percent:.1f}% used)")
|
|
except ImportError:
|
|
info_table.add_row("Memory", "Install psutil for memory info")
|
|
except Exception:
|
|
info_table.add_row("Memory", "Unable to determine")
|
|
|
|
console.print(info_table)
|
|
console.print()
|
|
|
|
# Helpful tips pointing to other commands
|
|
console.print(Panel(
|
|
"[dim]💡 For more information:[/dim]\n"
|
|
"• Run [cyan]tito system health[/cyan] for quick environment check\n"
|
|
"• Run [cyan]tito system version[/cyan] to see all package versions\n"
|
|
"• Run [cyan]tito system check[/cyan] for comprehensive validation (60+ tests)",
|
|
border_style="blue"
|
|
))
|
|
|
|
return 0
|