ml-dash 0.0.11__py3-none-any.whl → 0.5.9__py3-none-any.whl

ml_dash/__init__.py

@@ -1 +1,59 @@
-from .main import app
+"""
+ML-Dash Python SDK
+
+A simple and flexible SDK for ML experiment metricing and data storage.
+
+Usage:
+
+    # Remote mode (API server)
+    from ml_dash import Experiment
+
+    with Experiment(
+        name="my-experiment",
+        project="my-project",
+        remote="http://localhost:3000",
+        api_key="your-jwt-token"
+    ) as experiment:
+        experiment.log("Training started")
+        experiment.metric("loss", {"step": 0, "value": 0.5})
+
+    # Local mode (filesystem)
+    with Experiment(
+        name="my-experiment",
+        project="my-project",
+        local_path=".ml-dash"
+    ) as experiment:
+        experiment.log("Training started")
+
+    # Decorator style
+    from ml_dash import ml_dash_experiment
+
+    @ml_dash_experiment(
+        name="my-experiment",
+        project="my-project",
+        remote="http://localhost:3000",
+        api_key="your-jwt-token"
+    )
+    def train_model(experiment):
+        experiment.log("Training started")
+"""
+
+from .experiment import Experiment, ml_dash_experiment, OperationMode, RunManager
+from .client import RemoteClient
+from .storage import LocalStorage
+from .log import LogLevel, LogBuilder
+from .params import ParametersBuilder
+
+__version__ = "0.1.0"
+
+__all__ = [
+    "Experiment",
+    "ml_dash_experiment",
+    "OperationMode",
+    "RunManager",
+    "RemoteClient",
+    "LocalStorage",
+    "LogLevel",
+    "LogBuilder",
+    "ParametersBuilder",
+]

ml_dash/auto_start.py

@@ -0,0 +1,42 @@
+"""
+Auto-start module for ML-Dash SDK.
+
+Provides a pre-configured, auto-started experiment singleton named 'dxp'.
+
+Usage:
+    from ml_dash.auto_start import dxp
+
+    # Ready to use immediately - no need to open/start
+    dxp.log("Hello from dxp!")
+    dxp.params.set(lr=0.001)
+    dxp.metrics("loss").append(step=0, value=0.5)
+
+    # Automatically closed on Python exit
+"""
+
+import atexit
+from .experiment import Experiment
+
+# Create pre-configured singleton experiment
+dxp = Experiment(
+    name="dxp",
+    project="scratch",
+    local_path=".ml-dash"
+)
+
+# Auto-start the experiment on import
+dxp.run.start()
+
+# Register cleanup handler to complete experiment on Python exit
+def _cleanup():
+    """Complete the dxp experiment on exit if still open."""
+    if dxp._is_open:
+        try:
+            dxp.run.complete()
+        except Exception:
+            # Silently ignore errors during cleanup
+            pass
+
+atexit.register(_cleanup)
+
+__all__ = ["dxp"]

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/cli_commands/__init__.py

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