ml-dash 0.5.6__tar.gz → 0.5.8__tar.gz

{ml_dash-0.5.6 → ml_dash-0.5.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: ml-dash
-Version: 0.5.6
+Version: 0.5.8
 Summary: ML experiment tracking and data storage
 Keywords: machine-learning,experiment-tracking,mlops,data-storage
 Author: Ge Yang, Tom Tao
@@ -41,6 +41,7 @@ Requires-Dist: pyjwt>=2.8.0
 Requires-Dist: imageio>=2.31.0
 Requires-Dist: imageio-ffmpeg>=0.4.9
 Requires-Dist: scikit-image>=0.21.0
+Requires-Dist: rich>=13.0.0
 Requires-Dist: pytest>=8.0.0 ; extra == 'dev'
 Requires-Dist: pytest-asyncio>=0.23.0 ; extra == 'dev'
 Requires-Dist: sphinx>=7.2.0 ; extra == 'dev'

{ml_dash-0.5.6 → ml_dash-0.5.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ml-dash"
-version = "0.5.6"
+version = "0.5.8"
 description = "ML experiment tracking and data storage"
 readme = "README.md"
 requires-python = ">=3.9"
@@ -29,8 +29,12 @@ dependencies = [
     "imageio>=2.31.0",
     "imageio-ffmpeg>=0.4.9",
     "scikit-image>=0.21.0",
+    "rich>=13.0.0",
 ]
 
+[project.scripts]
+ml-dash = "ml_dash.cli:main"
+
 [project.optional-dependencies]
 dev = [
     "pytest>=8.0.0",

ml_dash-0.5.8/src/ml_dash/cli.py

@@ -0,0 +1,67 @@
+"""ML-Dash command-line interface."""
+
+import argparse
+import sys
+from typing import Optional, List
+
+
+def create_parser() -> argparse.ArgumentParser:
+    """Create the main CLI argument parser."""
+    parser = argparse.ArgumentParser(
+        prog="ml-dash",
+        description="ML-Dash: ML experiment tracking and data storage CLI",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+    )
+
+    # Add subcommands
+    subparsers = parser.add_subparsers(
+        dest="command",
+        help="Available commands",
+        metavar="COMMAND",
+    )
+
+    # Import and add command parsers
+    from .cli_commands import upload, download, list as list_cmd
+    upload.add_parser(subparsers)
+    download.add_parser(subparsers)
+    list_cmd.add_parser(subparsers)
+
+    return parser
+
+
+def main(argv: Optional[List[str]] = None) -> int:
+    """
+    Main CLI entry point.
+
+    Args:
+        argv: Command-line arguments (defaults to sys.argv[1:])
+
+    Returns:
+        Exit code (0 for success, non-zero for error)
+    """
+    parser = create_parser()
+    args = parser.parse_args(argv)
+
+    # If no command specified, show help
+    if args.command is None:
+        parser.print_help()
+        return 0
+
+    # Route to command handlers
+    if args.command == "upload":
+        from .cli_commands import upload
+        return upload.cmd_upload(args)
+    elif args.command == "download":
+        from .cli_commands import download
+        return download.cmd_download(args)
+    elif args.command == "list":
+        from .cli_commands import list as list_cmd
+        return list_cmd.cmd_list(args)
+
+    # Unknown command (shouldn't happen due to subparsers)
+    parser.print_help()
+    return 1
+
+
+if __name__ == "__main__":
+    sys.exit(main())

ml_dash-0.5.8/src/ml_dash/cli_commands/__init__.py

@@ -0,0 +1 @@
+"""CLI command modules for ML-Dash."""