mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-04-29 20:38:58 -05:00
e159a65118
- Fix AttributeError when calling reset without --all flag - Use getattr() with defaults for 'all' and 'list_backups' attributes
87 lines
3.9 KiB
Python
87 lines
3.9 KiB
Python
"""
|
|
nbdev command for TinyTorch CLI: runs nbdev commands for notebook development.
|
|
"""
|
|
|
|
import subprocess
|
|
from argparse import ArgumentParser, Namespace
|
|
from rich.panel import Panel
|
|
|
|
from .base import BaseCommand
|
|
|
|
class NbdevCommand(BaseCommand):
|
|
@property
|
|
def name(self) -> str:
|
|
return "nbdev"
|
|
|
|
@property
|
|
def description(self) -> str:
|
|
return "nbdev notebook development commands"
|
|
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
|
parser.add_argument("--export", action="store_true", help="Export notebooks to Python package")
|
|
parser.add_argument("--build-docs", action="store_true", help="Build documentation from notebooks")
|
|
parser.add_argument("--test", action="store_true", help="Run notebook tests")
|
|
parser.add_argument("--clean", action="store_true", help="Clean notebook outputs")
|
|
parser.add_argument("--all", action="store_true", help="Export all modules (use with --export)")
|
|
parser.add_argument("module", nargs="?", help="Export specific module (use with --export)")
|
|
|
|
def run(self, args: Namespace) -> int:
|
|
console = self.console
|
|
|
|
console.print(Panel("📓 nbdev Notebook Development",
|
|
title="Notebook Tools", border_style="bright_cyan"))
|
|
|
|
if args.export:
|
|
# Use the export command logic
|
|
from .export import ExportCommand
|
|
export_cmd = ExportCommand(self.config)
|
|
export_args = ArgumentParser()
|
|
export_cmd.add_arguments(export_args)
|
|
|
|
# Build the arguments for export command
|
|
export_arg_list = []
|
|
if args.all:
|
|
export_arg_list.append("--all")
|
|
elif args.module:
|
|
export_arg_list.append(args.module)
|
|
|
|
export_args = export_args.parse_args(export_arg_list)
|
|
return export_cmd.run(export_args)
|
|
|
|
elif args.build_docs:
|
|
console.print("📚 Building documentation from notebooks...")
|
|
result = subprocess.run(["nbdev_docs"], capture_output=True, text=True)
|
|
if result.returncode == 0:
|
|
console.print(Panel("[green]✅ Documentation built successfully![/green]",
|
|
title="Docs Success", border_style="green"))
|
|
else:
|
|
console.print(Panel(f"[red]❌ Docs build failed: {result.stderr}[/red]",
|
|
title="Docs Error", border_style="red"))
|
|
return result.returncode
|
|
|
|
elif args.test:
|
|
console.print("🧪 Running notebook tests...")
|
|
result = subprocess.run(["nbdev_test"], capture_output=True, text=True)
|
|
if result.returncode == 0:
|
|
console.print(Panel("[green]✅ Notebook tests passed![/green]",
|
|
title="Test Success", border_style="green"))
|
|
else:
|
|
console.print(Panel(f"[red]❌ Notebook tests failed: {result.stderr}[/red]",
|
|
title="Test Error", border_style="red"))
|
|
return result.returncode
|
|
|
|
elif args.clean:
|
|
console.print("🧹 Cleaning notebook outputs...")
|
|
result = subprocess.run(["nbdev_clean"], capture_output=True, text=True)
|
|
if result.returncode == 0:
|
|
console.print(Panel("[green]✅ Notebook outputs cleaned![/green]",
|
|
title="Clean Success", border_style="green"))
|
|
else:
|
|
console.print(Panel(f"[red]❌ Clean failed: {result.stderr}[/red]",
|
|
title="Clean Error", border_style="red"))
|
|
return result.returncode
|
|
|
|
else:
|
|
console.print(Panel("[yellow]⚠️ No nbdev action specified. Use --export, --build-docs, --test, or --clean[/yellow]",
|
|
title="No Action", border_style="yellow"))
|
|
return 1 |