ml-dash 0.6.2__py3-none-any.whl → 0.6.2rc1__py3-none-any.whl

ml_dash/auto_start.py

@@ -9,56 +9,43 @@ Usage:
     # First, authenticate
     # $ ml-dash login
 
-    from ml_dash.auto_start import dxp
+    from ml_dash import dxp
 
     # Use with statement (recommended)
     with dxp.run:
-        dxp.log("Hello from dxp!", level="info")
+        dxp.log().info("Hello from dxp!")
         dxp.params.set(lr=0.001)
-        dxp.metrics("train").log(loss=0.5, step=0)
+        dxp.metrics("loss").append(step=0, value=0.5)
     # Automatically completes on exit from with block
 
     # Or start/complete manually
     dxp.run.start()
-    dxp.log("Training...", level="info")
+    dxp.log().info("Training...")
     dxp.run.complete()
 """
 
 import atexit
+from .experiment import Experiment
 
 # Create pre-configured singleton experiment in remote mode
 # Uses default remote server (https://api.dash.ml)
 # Token is auto-loaded from storage when first used
 # If not authenticated, operations will fail with AuthenticationError
-# Prefix format: {owner}/{project}/path...
-# Using getpass to get current user as owner for local convenience
-import getpass
-from datetime import datetime
-
-from .auth.token_storage import get_jwt_user
-from .experiment import Experiment
-
-_user = get_jwt_user()
-# Fallback to system username if not authenticated
-_username = _user["username"] if _user else getpass.getuser()
-_now = datetime.now()
-
 dxp = Experiment(
-  prefix=f"{_username}/scratch/{_now:%Y-%m-%d/%H%M%S}",
-  dash_url="https://api.dash.ml",
+    name="dxp",
+    project="scratch",
+    remote="https://api.dash.ml",
 )
 
-
 # Register cleanup handler to complete experiment on Python exit (if still open)
 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
-
+    """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)
 

ml_dash/cli.py

@@ -9,11 +9,7 @@ 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\n\n"
-            "View your experiments, statistics, and plots online at:\n"
-            "  https://dash.ml\n"
-        ),
+        description="ML-Dash: ML experiment tracking and data storage CLI",
         formatter_class=argparse.RawDescriptionHelpFormatter,
     )
 
@@ -25,15 +21,11 @@ def create_parser() -> argparse.ArgumentParser:
     )
 
     # Import and add command parsers
-    from .cli_commands import upload, download, list as list_cmd, login, logout, profile, api
+    from .cli_commands import upload, download, list as list_cmd, login, logout
 
     # Authentication commands
     login.add_parser(subparsers)
     logout.add_parser(subparsers)
-    profile.add_parser(subparsers)
-
-    # API commands
-    api.add_parser(subparsers)
 
     # Data commands
     upload.add_parser(subparsers)
@@ -68,9 +60,6 @@ def main(argv: Optional[List[str]] = None) -> int:
     elif args.command == "logout":
         from .cli_commands import logout
         return logout.cmd_logout(args)
-    elif args.command == "profile":
-        from .cli_commands import profile
-        return profile.cmd_profile(args)
     elif args.command == "upload":
         from .cli_commands import upload
         return upload.cmd_upload(args)
@@ -80,9 +69,6 @@ def main(argv: Optional[List[str]] = None) -> int:
     elif args.command == "list":
         from .cli_commands import list as list_cmd
         return list_cmd.cmd_list(args)
-    elif args.command == "api":
-        from .cli_commands import api
-        return api.cmd_api(args)
 
     # Unknown command (shouldn't happen due to subparsers)
     parser.print_help()