mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-05-25 15:09:54 -05:00
a92a5530ef
BREAKING CHANGE: CLI moved from tinytorch/cli/ to tito/ Perfect Senior Engineer Architecture: - tinytorch/ = Pure ML framework (production) - tito/ = Development/management CLI tool - modules/ = Educational content Benefits: ✅ Clean separation of concerns ✅ Framework stays lightweight (no CLI dependencies) ✅ Clear mental model for users ✅ Professional project organization ✅ Proper dependency management Structure: tinytorch/ # 🧠 Core ML Framework ├── core/ # Tensors, layers, operations ├── training/ # Training loops, optimizers ├── models/ # Model architectures └── ... # Pure ML functionality tito/ # 🔧 Development CLI Tool ├── main.py # CLI entry point ├── core/ # CLI configuration & console ├── commands/ # Command implementations └── tools/ # CLI utilities Key Changes: - Moved all CLI code from tinytorch/cli/ to tito/ - Updated imports and entry points - Separated dependencies (Rich only for dev tools) - Updated documentation to reflect proper separation - Maintained backward compatibility with bin/tito wrapper This demonstrates how senior engineers separate: - Production code (framework) from development tools (CLI) - Core functionality from management utilities - User-facing APIs from internal tooling Educational Value: - Shows proper software architecture - Teaches separation of concerns - Demonstrates dependency management - Models real-world project organization
48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
"""
|
||
Console management for consistent CLI output.
|
||
"""
|
||
|
||
from rich.console import Console
|
||
from rich.panel import Panel
|
||
from rich.text import Text
|
||
from rich.table import Table
|
||
from rich.tree import Tree
|
||
from rich.progress import Progress, SpinnerColumn, BarColumn, TextColumn
|
||
from typing import Optional
|
||
import sys
|
||
|
||
# Global console instance
|
||
_console: Optional[Console] = None
|
||
|
||
def get_console() -> Console:
|
||
"""Get the global console instance."""
|
||
global _console
|
||
if _console is None:
|
||
_console = Console(stderr=False)
|
||
return _console
|
||
|
||
def print_banner():
|
||
"""Print the TinyTorch banner using Rich."""
|
||
console = get_console()
|
||
banner_text = Text("Tiny🔥Torch: Build ML Systems from Scratch", style="bold red")
|
||
console.print(Panel(banner_text, style="bright_blue", padding=(1, 2)))
|
||
|
||
def print_error(message: str, title: str = "Error"):
|
||
"""Print an error message with consistent formatting."""
|
||
console = get_console()
|
||
console.print(Panel(f"[red]❌ {message}[/red]", title=title, border_style="red"))
|
||
|
||
def print_success(message: str, title: str = "Success"):
|
||
"""Print a success message with consistent formatting."""
|
||
console = get_console()
|
||
console.print(Panel(f"[green]✅ {message}[/green]", title=title, border_style="green"))
|
||
|
||
def print_warning(message: str, title: str = "Warning"):
|
||
"""Print a warning message with consistent formatting."""
|
||
console = get_console()
|
||
console.print(Panel(f"[yellow]⚠️ {message}[/yellow]", title=title, border_style="yellow"))
|
||
|
||
def print_info(message: str, title: str = "Info"):
|
||
"""Print an info message with consistent formatting."""
|
||
console = get_console()
|
||
console.print(Panel(f"[cyan]ℹ️ {message}[/cyan]", title=title, border_style="cyan")) |