github-starred/TinyTorch
2
0
Fork 0
mirror of https://github.com/MLSysBook/TinyTorch.git synced 2026-03-11 20:55:19 -05:00
Code Issues 7 Packages Projects Releases Wiki Activity

Adds module-specific sync functionality

Browse Source 
Extends the sync command to allow users to synchronize
specific modules instead of the entire project. This
improves efficiency by reducing the scope of the nbdev
export process. Adds argument parsing for module selection.
This commit is contained in:
Vijay Janapa Reddi
2025-07-10 12:47:17 -04:00
parent 5e75d042e5
commit 1ac530d8ff
1 changed files with 23 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
bin/tito.py
View File

@@ -558,13 +558,30 @@ def cmd_sync(args):
"""Export notebook code to Python package using nbdev."""
import subprocess
console.print(Panel("🔄 Synchronizing Notebooks to Package",
title="nbdev Export", border_style="bright_cyan"))
console.print("🔄 Exporting notebook code to tinytorch package...")
# Determine what to sync
if hasattr(args, 'module') and args.module:
module_path = f"modules/{args.module}"
if not Path(module_path).exists():
console.print(Panel(f"[red]❌ Module '{args.module}' not found at {module_path}[/red]",
title="Module Not Found", border_style="red"))
return 1
console.print(Panel(f"🔄 Synchronizing Module: {args.module}",
title="nbdev Export", border_style="bright_cyan"))
console.print(f"🔄 Exporting {args.module} notebook to tinytorch package...")
# Use nbdev_export with --path for specific module
cmd = ["nbdev_export", "--path", module_path]
else:
console.print(Panel("🔄 Synchronizing All Notebooks to Package",
title="nbdev Export", border_style="bright_cyan"))
console.print("🔄 Exporting all notebook code to tinytorch package...")
# Use nbdev_export for all modules
cmd = ["nbdev_export"]
try:
result = subprocess.run(["nbdev_export"], capture_output=True, text=True, cwd=Path.cwd())
result = subprocess.run(cmd, capture_output=True, text=True, cwd=Path.cwd())
if result.returncode == 0:
console.print(Panel("[green]✅ Successfully exported notebook code to tinytorch package![/green]",
@@ -706,6 +723,7 @@ def main():
# Sync command (primary nbdev export)
sync_parser = subparsers.add_parser("sync", help="Export notebook code to Python package")
sync_parser.add_argument("--module", help="Sync specific module (e.g., setup, tensor)")
# nbdev commands
nbdev_parser = subparsers.add_parser("nbdev", help="nbdev notebook development commands")