podstack 1.3.13__tar.gz → 1.3.15__tar.gz

{podstack-1.3.13 → podstack-1.3.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: podstack
-Version: 1.3.13
+Version: 1.3.15
 Summary: Official Python SDK for Podstack GPU Notebook Platform
 Author-email: Podstack <support@podstack.ai>
 License-Expression: MIT
@@ -298,9 +298,9 @@ with registry.start_run(name="training-v1") as run:
     # Set tags
     registry.set_tag("framework", "pytorch")
 
-    # Log artifacts
-    registry.log_artifact("model.pt", "model")
-    registry.log_artifact("training_curves.png", "plots")
+    # Upload artifacts to cloud artifact store
+    registry.log_artifact("model.pt")
+    registry.log_artifact("training_curves.png", artifact_path="plots/curves.png")
 
     # Log dataset provenance (first-class resource, deduped by content hash)
     registry.log_dataset("imdb-reviews", path="data/imdb.csv", context="training")
@@ -316,7 +316,7 @@ with registry.start_run(name="training-v1") as run:
 ```python
 from podstack import registry
 
-# Log a model object (auto-detects framework)
+# Serialize and upload the model to the artifact store (auto-detects framework)
 registry.log_model(model, artifact_path="model", framework="pytorch")
 
 # Register in model registry
@@ -332,7 +332,7 @@ registry.set_model_stage("my-classifier", version=1, stage="production")
 # Set aliases
 registry.set_model_alias("my-classifier", alias="champion", version=1)
 
-# Load model from registry
+# Load model from any machine — files are downloaded automatically if missing locally
 model = registry.load_model("my-classifier", stage="production")
 ```
 
@@ -659,6 +659,155 @@ model = registry.get_model("sentiment-bert")
 lineage = registry.get_model_lineage(model.id)
 ```
 
+### Artifact Storage
+
+Podstack stores every artifact you log — model files, plots, CSV exports, anything — in the project's cloud artifact store. Artifacts are keyed by run ID, so the same file can be retrieved from any machine, by any project member, at any time.
+
+#### `log_artifact()` — upload a file for the active run
+
+```python
+# Upload a single file (uses the filename as the artifact path)
+registry.log_artifact("model.pt")
+
+# Upload with an explicit path inside the artifact store
+registry.log_artifact("training_curves.png", artifact_path="plots/curves.png")
+registry.log_artifact("feature_importance.csv", artifact_path="analysis/features.csv")
+```
+
+**Parameters:**
+
+| Parameter | Type | Default | Description |
+|-----------|------|---------|-------------|
+| `local_path` | `str` | required | Path to the local file to upload |
+| `artifact_path` | `str` | filename | Relative path inside the artifact store. Defaults to `os.path.basename(local_path)` |
+
+If the artifact store is temporarily unreachable, the SDK saves the file to a local fallback cache (`~/.podstack/artifacts/<run_id>/`) so your run is never interrupted.
+
+**Via the `Run` object** — equivalent to calling `registry.log_artifact()`:
+
+```python
+with registry.start_run("training-v1") as run:
+    run.log_artifact("confusion_matrix.png", artifact_path="plots/confusion_matrix.png")
+    run.log_artifact("model.pkl")
+```
+
+#### `list_artifacts()` — list all artifacts for a run
+
+```python
+artifacts = registry.list_artifacts(run_id)
+for a in artifacts:
+    print(f"{a['path']:40s}  {a['size'] / 1e6:.1f} MB  {a['last_modified']}")
+```
+
+**Parameters:**
+
+| Parameter | Type | Description |
+|-----------|------|-------------|
+| `run_id` | `str` | ID of the run to query |
+
+**Returns:** `list[dict]` — one entry per artifact:
+
+| Key | Type | Description |
+|-----|------|-------------|
+| `path` | `str` | Relative artifact path (e.g. `"plots/curves.png"`) |
+| `size` | `int` | File size in bytes |
+| `etag` | `str` | Content hash for integrity verification |
+| `last_modified` | `str` | ISO 8601 upload timestamp |
+
+#### `download_artifact()` — retrieve an artifact
+
+Downloads a specific artifact from the cloud store into a local directory. Falls back to the local cache when the store is unreachable.
+
+```python
+# Download a single file
+dest = registry.download_artifact("run-id", "model/model.pkl", "./downloads/")
+print(f"Saved to: {dest}")
+
+# Download a whole model directory
+dest = registry.download_artifact("run-id", "model", "./local_models/")
+```
+
+**Parameters:**
+
+| Parameter | Type | Description |
+|-----------|------|-------------|
+| `run_id` | `str` | ID of the run that logged the artifact |
+| `artifact_path` | `str` | Relative artifact path as logged (e.g. `"model/model.pkl"`) |
+| `local_path` | `str` | Destination directory |
+
+**Returns:** `str` — absolute path to the downloaded file or directory.
+
+**Raises:** `ArtifactNotFoundError` if the artifact cannot be found in the store or the local cache.
+
+#### Models as artifacts: `log_model()` and `load_model()`
+
+`log_model()` serializes your model to disk and uploads every resulting file to the artifact store in one call. `load_model()` resolves the registered model version, downloads any missing files from the store, then deserializes the model — so it works correctly from any machine regardless of where training happened.
+
+```python
+# ── Training machine ──────────────────────────────────────────────────────────
+with registry.start_run("bert-finetune-v3") as run:
+    # train...
+    registry.log_model(model, artifact_path="model", framework="pytorch")
+
+registry.register_model("sentiment-bert", run_id=run.id)
+registry.set_model_stage("sentiment-bert", version=3, stage="production")
+
+# ── Any machine (CI, inference server, colleague's laptop) ───────────────────
+# Model files are downloaded automatically from the artifact store if not cached
+model = registry.load_model("sentiment-bert", stage="production")
+```
+
+**`log_model()` parameters:**
+
+| Parameter | Type | Default | Description |
+|-----------|------|---------|-------------|
+| `model` | any | required | Model object (PyTorch, TensorFlow, sklearn, HuggingFace, or any picklable object) |
+| `artifact_path` | `str` | `"model"` | Sub-path inside the artifact store |
+| `framework` | `str` | auto-detected | `"pytorch"`, `"tensorflow"`, `"sklearn"`, `"huggingface"`, or `"pickle"` |
+| `metadata` | `dict` | `None` | Arbitrary key-value metadata stored as run params |
+
+**`load_model()` parameters:**
+
+| Parameter | Type | Default | Description |
+|-----------|------|---------|-------------|
+| `model_name` | `str` | required | Registered model name |
+| `version` | `int` | `None` | Specific version to load. Mutually exclusive with `stage` |
+| `stage` | `str` | `None` | Stage to load from: `"development"`, `"staging"`, `"production"`, `"archived"` |
+| `framework` | `str` | from run params | Override framework for deserialization |
+
+#### Viewing artifacts in the dashboard
+
+Every artifact logged with `log_artifact()` or `log_model()` appears automatically in the **Artifacts tab** of the run's detail page in the Podstack dashboard. No extra steps are needed — the tab populates from the same store the SDK writes to.
+
+The Artifacts tab shows:
+
+| Column | Description |
+|--------|-------------|
+| **Path** | The relative artifact path as logged (e.g. `model/model.pkl`, `plots/curves.png`) |
+| **Type badge** | File extension, color-coded by category — model weights, data files, images, configs, etc. |
+| **Size** | Formatted file size (B / KB / MB) |
+| **Uploaded** | Timestamp of when the file was stored |
+| **Download** | One-click download button — opens a short-lived direct download link in the browser |
+
+A footer below the list shows the combined size of all artifacts for the run.
+
+```python
+# Everything logged here shows up in the dashboard Artifacts tab
+with registry.start_run("bert-finetune-v3") as run:
+    registry.log_params({"lr": 2e-5, "epochs": 3})
+    registry.log_metrics({"accuracy": 0.93})
+
+    # These all appear as separate rows in the Artifacts tab
+    registry.log_artifact("confusion_matrix.png", artifact_path="plots/confusion_matrix.png")
+    registry.log_artifact("feature_importance.csv", artifact_path="analysis/features.csv")
+    registry.log_model(model, artifact_path="model", framework="pytorch")
+    # ↳ each model file (model.pkl, config.json, etc.) appears as its own row
+```
+
+#### Access control
+
+Artifact upload and download URLs are issued by the registry API and require a valid API key and project membership. The URLs are short-lived, ensuring that access always reflects the current state of your project — a revoked key can no longer generate new URLs. Any member of a project can upload and download artifacts for runs within that project.
+
 ### List and Browse
 
 ```python
@@ -670,8 +819,12 @@ experiments = registry.list_experiments()
 # List models
 models = registry.list_models()
 
-# Download artifacts
-registry.download_artifact("run-id", "model/model.pt", "./downloads/")
+# List artifacts for a specific run
+artifacts = registry.list_artifacts(run_id)
+
+# Download a specific artifact to a local directory
+dest = registry.download_artifact("run-id", "model/model.pt", "./downloads/")
+print(f"Saved to: {dest}")
 ```
 
 ## GPU Runner - Direct Code Execution

{podstack-1.3.13 → podstack-1.3.15}/README.md

@@ -246,9 +246,9 @@ with registry.start_run(name="training-v1") as run:
     # Set tags
     registry.set_tag("framework", "pytorch")
 
-    # Log artifacts
-    registry.log_artifact("model.pt", "model")
-    registry.log_artifact("training_curves.png", "plots")
+    # Upload artifacts to cloud artifact store
+    registry.log_artifact("model.pt")
+    registry.log_artifact("training_curves.png", artifact_path="plots/curves.png")
 
     # Log dataset provenance (first-class resource, deduped by content hash)
     registry.log_dataset("imdb-reviews", path="data/imdb.csv", context="training")
@@ -264,7 +264,7 @@ with registry.start_run(name="training-v1") as run:
 ```python
 from podstack import registry
 
-# Log a model object (auto-detects framework)
+# Serialize and upload the model to the artifact store (auto-detects framework)
 registry.log_model(model, artifact_path="model", framework="pytorch")
 
 # Register in model registry
@@ -280,7 +280,7 @@ registry.set_model_stage("my-classifier", version=1, stage="production")
 # Set aliases
 registry.set_model_alias("my-classifier", alias="champion", version=1)
 
-# Load model from registry
+# Load model from any machine — files are downloaded automatically if missing locally
 model = registry.load_model("my-classifier", stage="production")
 ```
 
@@ -607,6 +607,155 @@ model = registry.get_model("sentiment-bert")
 lineage = registry.get_model_lineage(model.id)
 ```
 
+### Artifact Storage
+
+Podstack stores every artifact you log — model files, plots, CSV exports, anything — in the project's cloud artifact store. Artifacts are keyed by run ID, so the same file can be retrieved from any machine, by any project member, at any time.
+
+#### `log_artifact()` — upload a file for the active run
+
+```python
+# Upload a single file (uses the filename as the artifact path)
+registry.log_artifact("model.pt")
+
+# Upload with an explicit path inside the artifact store
+registry.log_artifact("training_curves.png", artifact_path="plots/curves.png")
+registry.log_artifact("feature_importance.csv", artifact_path="analysis/features.csv")
+```
+
+**Parameters:**
+
+| Parameter | Type | Default | Description |
+|-----------|------|---------|-------------|
+| `local_path` | `str` | required | Path to the local file to upload |
+| `artifact_path` | `str` | filename | Relative path inside the artifact store. Defaults to `os.path.basename(local_path)` |
+
+If the artifact store is temporarily unreachable, the SDK saves the file to a local fallback cache (`~/.podstack/artifacts/<run_id>/`) so your run is never interrupted.
+
+**Via the `Run` object** — equivalent to calling `registry.log_artifact()`:
+
+```python
+with registry.start_run("training-v1") as run:
+    run.log_artifact("confusion_matrix.png", artifact_path="plots/confusion_matrix.png")
+    run.log_artifact("model.pkl")
+```
+
+#### `list_artifacts()` — list all artifacts for a run
+
+```python
+artifacts = registry.list_artifacts(run_id)
+for a in artifacts:
+    print(f"{a['path']:40s}  {a['size'] / 1e6:.1f} MB  {a['last_modified']}")
+```
+
+**Parameters:**
+
+| Parameter | Type | Description |
+|-----------|------|-------------|
+| `run_id` | `str` | ID of the run to query |
+
+**Returns:** `list[dict]` — one entry per artifact:
+
+| Key | Type | Description |
+|-----|------|-------------|
+| `path` | `str` | Relative artifact path (e.g. `"plots/curves.png"`) |
+| `size` | `int` | File size in bytes |
+| `etag` | `str` | Content hash for integrity verification |
+| `last_modified` | `str` | ISO 8601 upload timestamp |
+
+#### `download_artifact()` — retrieve an artifact
+
+Downloads a specific artifact from the cloud store into a local directory. Falls back to the local cache when the store is unreachable.
+
+```python
+# Download a single file
+dest = registry.download_artifact("run-id", "model/model.pkl", "./downloads/")
+print(f"Saved to: {dest}")
+
+# Download a whole model directory
+dest = registry.download_artifact("run-id", "model", "./local_models/")
+```
+
+**Parameters:**
+
+| Parameter | Type | Description |
+|-----------|------|-------------|
+| `run_id` | `str` | ID of the run that logged the artifact |
+| `artifact_path` | `str` | Relative artifact path as logged (e.g. `"model/model.pkl"`) |
+| `local_path` | `str` | Destination directory |
+
+**Returns:** `str` — absolute path to the downloaded file or directory.
+
+**Raises:** `ArtifactNotFoundError` if the artifact cannot be found in the store or the local cache.
+
+#### Models as artifacts: `log_model()` and `load_model()`
+
+`log_model()` serializes your model to disk and uploads every resulting file to the artifact store in one call. `load_model()` resolves the registered model version, downloads any missing files from the store, then deserializes the model — so it works correctly from any machine regardless of where training happened.
+
+```python
+# ── Training machine ──────────────────────────────────────────────────────────
+with registry.start_run("bert-finetune-v3") as run:
+    # train...
+    registry.log_model(model, artifact_path="model", framework="pytorch")
+
+registry.register_model("sentiment-bert", run_id=run.id)
+registry.set_model_stage("sentiment-bert", version=3, stage="production")
+
+# ── Any machine (CI, inference server, colleague's laptop) ───────────────────
+# Model files are downloaded automatically from the artifact store if not cached
+model = registry.load_model("sentiment-bert", stage="production")
+```
+
+**`log_model()` parameters:**
+
+| Parameter | Type | Default | Description |
+|-----------|------|---------|-------------|
+| `model` | any | required | Model object (PyTorch, TensorFlow, sklearn, HuggingFace, or any picklable object) |
+| `artifact_path` | `str` | `"model"` | Sub-path inside the artifact store |
+| `framework` | `str` | auto-detected | `"pytorch"`, `"tensorflow"`, `"sklearn"`, `"huggingface"`, or `"pickle"` |
+| `metadata` | `dict` | `None` | Arbitrary key-value metadata stored as run params |
+
+**`load_model()` parameters:**
+
+| Parameter | Type | Default | Description |
+|-----------|------|---------|-------------|
+| `model_name` | `str` | required | Registered model name |
+| `version` | `int` | `None` | Specific version to load. Mutually exclusive with `stage` |
+| `stage` | `str` | `None` | Stage to load from: `"development"`, `"staging"`, `"production"`, `"archived"` |
+| `framework` | `str` | from run params | Override framework for deserialization |
+
+#### Viewing artifacts in the dashboard
+
+Every artifact logged with `log_artifact()` or `log_model()` appears automatically in the **Artifacts tab** of the run's detail page in the Podstack dashboard. No extra steps are needed — the tab populates from the same store the SDK writes to.
+
+The Artifacts tab shows:
+
+| Column | Description |
+|--------|-------------|
+| **Path** | The relative artifact path as logged (e.g. `model/model.pkl`, `plots/curves.png`) |
+| **Type badge** | File extension, color-coded by category — model weights, data files, images, configs, etc. |
+| **Size** | Formatted file size (B / KB / MB) |
+| **Uploaded** | Timestamp of when the file was stored |
+| **Download** | One-click download button — opens a short-lived direct download link in the browser |
+
+A footer below the list shows the combined size of all artifacts for the run.
+
+```python
+# Everything logged here shows up in the dashboard Artifacts tab
+with registry.start_run("bert-finetune-v3") as run:
+    registry.log_params({"lr": 2e-5, "epochs": 3})
+    registry.log_metrics({"accuracy": 0.93})
+
+    # These all appear as separate rows in the Artifacts tab
+    registry.log_artifact("confusion_matrix.png", artifact_path="plots/confusion_matrix.png")
+    registry.log_artifact("feature_importance.csv", artifact_path="analysis/features.csv")
+    registry.log_model(model, artifact_path="model", framework="pytorch")
+    # ↳ each model file (model.pkl, config.json, etc.) appears as its own row
+```
+
+#### Access control
+
+Artifact upload and download URLs are issued by the registry API and require a valid API key and project membership. The URLs are short-lived, ensuring that access always reflects the current state of your project — a revoked key can no longer generate new URLs. Any member of a project can upload and download artifacts for runs within that project.
+
 ### List and Browse
 
 ```python
@@ -618,8 +767,12 @@ experiments = registry.list_experiments()
 # List models
 models = registry.list_models()
 
-# Download artifacts
-registry.download_artifact("run-id", "model/model.pt", "./downloads/")
+# List artifacts for a specific run
+artifacts = registry.list_artifacts(run_id)
+
+# Download a specific artifact to a local directory
+dest = registry.download_artifact("run-id", "model/model.pt", "./downloads/")
+print(f"Saved to: {dest}")
 ```
 
 ## GPU Runner - Direct Code Execution

{podstack-1.3.13 → podstack-1.3.15}/podstack/registry/__init__.py

@@ -47,18 +47,22 @@ __all__ = [
     "set_experiment",
     "get_experiment",
     "list_experiments",
+    "archive_experiment",
     "start_run",
     "end_run",
+    "get_run",
+    "list_runs",
     "log_params",
     "log_metrics",
     "log_artifact",
     "set_tag",
+    "update_run_notes",
     "register_model",
     "get_model",
     "list_models",
     "set_model_stage",
     "set_model_alias",
-    # New methods
+    # MLOps helpers
     "log_model",
     "load_model",
     "log_dataset",
@@ -66,6 +70,32 @@ __all__ = [
     "get_metric_history",
     "download_artifact",
     "search_runs",
+    "get_run_datasets",
+    "get_model_lineage",
+    "autolog",
+    # HPO Sweeps
+    "create_sweep",
+    "get_sweep",
+    "list_sweeps",
+    "suggest_trial_params",
+    "create_trial",
+    "complete_trial",
+    "list_trials",
+    "stop_sweep",
+    # Alerts
+    "create_alert",
+    "list_alerts",
+    "delete_alert",
+    # Approvals
+    "list_pending_approvals",
+    "approve_promotion",
+    "reject_promotion",
+    # Schedules
+    "create_schedule",
+    "get_schedule",
+    "update_schedule",
+    "delete_schedule",
+    "list_schedules",
     # Classes
     "Experiment",
     "Run",
@@ -400,3 +430,227 @@ def search_runs(
         List of matching Run objects.
     """
     return _get_client().search_runs(experiment_id, status, max_results, offset)
+
+
+def get_run(run_id: str):
+    """Get a run by ID."""
+    return _get_client().get_run(run_id)
+
+
+def list_runs(
+    experiment_id: str = None,
+    status: str = None,
+    limit: int = 20,
+    offset: int = 0
+) -> list:
+    """List runs, optionally filtered by experiment or status."""
+    return _get_client().list_runs(experiment_id, status, limit, offset)
+
+
+def archive_experiment(experiment_id: str):
+    """Archive an experiment."""
+    return _get_client().archive_experiment(experiment_id)
+
+
+def update_run_notes(run_id: str, notes: str):
+    """Update the free-form notes for a run."""
+    _get_client().update_run_notes(run_id, notes)
+
+
+def get_run_datasets(run_id: str) -> list:
+    """List datasets logged for a run."""
+    return _get_client().get_run_datasets(run_id)
+
+
+def get_model_lineage(model_id: str) -> dict:
+    """Get full lineage for a model (versions → runs → datasets)."""
+    return _get_client().get_model_lineage(model_id)
+
+
+def autolog(
+    framework: str = None,
+    log_every_n_steps: int = 1,
+    log_system_metrics: bool = True,
+    system_metrics_interval: float = 10.0,
+):
+    """
+    Enable automatic logging for ML training frameworks.
+
+    Supports pytorch_lightning, huggingface, and sklearn.
+    Auto-detects available frameworks when framework is None.
+    """
+    _get_client().autolog(framework, log_every_n_steps, log_system_metrics, system_metrics_interval)
+
+
+# ==================== HPO Sweeps ====================
+
+def create_sweep(
+    experiment_id: str,
+    name: str,
+    search_space: dict,
+    strategy: str = "random",
+    max_trials: int = 20,
+    metric=None,
+    direction: str = "minimize",
+) -> dict:
+    """
+    Create a hyperparameter optimization sweep.
+
+    Args:
+        experiment_id: Experiment to run trials in.
+        name: Sweep name.
+        search_space: Dict mapping param names to spec dicts.
+        strategy: "random" (default) or "grid".
+        max_trials: Maximum number of trials.
+        metric: Metric key to optimize (str), or dict with "name" and "direction" keys.
+        direction: "minimize" (default) or "maximize". Ignored if metric is a dict.
+
+    Returns:
+        Sweep dict with id, status, etc.
+    """
+    if isinstance(metric, dict):
+        direction = metric.get("direction", direction)
+        metric = metric.get("name", None)
+    return _get_client().create_sweep(
+        experiment_id, name, search_space, strategy, max_trials, metric, direction
+    )
+
+
+def get_sweep(sweep_id: str) -> dict:
+    """Get a sweep by ID."""
+    return _get_client().get_sweep(sweep_id)
+
+
+def list_sweeps(experiment_id: str) -> list:
+    """List all sweeps for an experiment."""
+    return _get_client().list_sweeps(experiment_id)
+
+
+def suggest_trial_params(sweep_id: str) -> dict:
+    """Get suggested hyperparameter values for the next trial."""
+    return _get_client().suggest_trial_params(sweep_id)
+
+
+def create_trial(sweep_id: str, run_id: str, params: dict) -> dict:
+    """Record a new trial linked to a sweep and run."""
+    return _get_client().create_trial(sweep_id, run_id, params)
+
+
+def complete_trial(sweep_id: str, trial_id: str, value: float) -> None:
+    """Mark a trial as completed with its objective metric value."""
+    _get_client().complete_trial(sweep_id, trial_id, value)
+
+
+def list_trials(sweep_id: str) -> list:
+    """List all trials for a sweep."""
+    return _get_client().list_trials(sweep_id)
+
+
+def stop_sweep(sweep_id: str) -> None:
+    """Stop a running sweep."""
+    _get_client().stop_sweep(sweep_id)
+
+
+# ==================== Alerts ====================
+
+def create_alert(
+    run_id: str,
+    metric_key: str,
+    condition: str,
+    threshold: float,
+    notify_email: str = None,
+    notify_slack: str = None,
+) -> dict:
+    """
+    Create a metric threshold alert for a run.
+
+    Args:
+        run_id: Run to monitor.
+        metric_key: Metric name to watch.
+        condition: One of gt, lt, gte, lte, eq.
+        threshold: Trigger threshold value.
+        notify_email: Email address to notify.
+        notify_slack: Slack webhook URL to notify.
+
+    Returns:
+        Alert dict with id.
+    """
+    return _get_client().create_alert(run_id, metric_key, condition, threshold, notify_email, notify_slack)
+
+
+def list_alerts(run_id: str) -> list:
+    """List all alerts for a run."""
+    return _get_client().list_alerts(run_id)
+
+
+def delete_alert(alert_id: str) -> None:
+    """Delete an alert by ID."""
+    _get_client().delete_alert(alert_id)
+
+
+# ==================== Approvals ====================
+
+def list_pending_approvals() -> list:
+    """List all pending model promotion approval requests in the project."""
+    return _get_client().list_pending_approvals()
+
+
+def approve_promotion(request_id: str, comment: str = None) -> dict:
+    """Approve a pending model stage promotion request."""
+    return _get_client().approve_promotion(request_id, comment)
+
+
+def reject_promotion(request_id: str, comment: str = None) -> dict:
+    """Reject a pending model stage promotion request."""
+    return _get_client().reject_promotion(request_id, comment)
+
+
+# ==================== Schedules ====================
+
+def create_schedule(
+    name: str,
+    experiment_id: str,
+    cron_expr: str,
+    run_name: str = None,
+    run_config: dict = None,
+    webhook_url: str = None,
+) -> dict:
+    """
+    Create a recurring training schedule using a cron expression.
+
+    Args:
+        name: Schedule name.
+        experiment_id: Experiment to create runs in.
+        cron_expr: 5-field cron expression (e.g. "0 2 * * 1").
+        run_name: Base name for created runs.
+        run_config: Optional params to log on each scheduled run.
+        webhook_url: Optional URL to POST after each run fires.
+
+    Returns:
+        Schedule dict with id, next_fire_at, etc.
+    """
+    return _get_client().create_schedule(name, experiment_id, cron_expr, run_name, run_config, webhook_url)
+
+
+def list_schedules() -> list:
+    """List all training schedules in the project."""
+    return _get_client().list_schedules()
+
+
+def get_schedule(schedule_id: str) -> dict:
+    """Get a schedule by ID."""
+    return _get_client().get_schedule(schedule_id)
+
+
+def update_schedule(
+    schedule_id: str,
+    enabled: bool = None,
+    cron_expr: str = None,
+) -> dict:
+    """Update a schedule's enabled state or cron expression."""
+    return _get_client().update_schedule(schedule_id, enabled, cron_expr)
+
+
+def delete_schedule(schedule_id: str) -> None:
+    """Delete a schedule."""
+    _get_client().delete_schedule(schedule_id)