calkit-python 0.34.5__tar.gz → 0.34.6__tar.gz

{calkit_python-0.34.5 → calkit_python-0.34.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: calkit-python
-Version: 0.34.5
+Version: 0.34.6
 Summary: Reproducibility simplified.
 Project-URL: Homepage, https://calkit.org
 Project-URL: Issues, https://github.com/calkit/calkit/issues

{calkit_python-0.34.5 → calkit_python-0.34.6}/calkit/cli/list.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Literal
+from typing import Annotated, Literal
 
 import typer
 
@@ -125,3 +125,18 @@ def list_releases():
             typer.echo(f"    published: {published}")
         for k, v in obj.items():
             typer.echo(f"    {k}: {v}")
+
+
+@list_app.command(name="stages")
+def list_stages(
+    kinds: Annotated[
+        list[str] | None,
+        typer.Option("--kind", "-k", help="Filter stages by kind."),
+    ] = None,
+):
+    """List stages."""
+    stages = calkit.load_calkit_info().get("pipeline", {}).get("stages", {})
+    for name, stage in stages.items():
+        if kinds is not None and stage.get("kind") not in kinds:
+            continue
+        typer.echo(name)

calkit_python-0.34.6/calkit/tests/cli/test_list.py

@@ -0,0 +1,36 @@
+"""Tests for ``cli.list``."""
+
+import subprocess
+
+import calkit
+
+
+def test_list_environments(tmp_dir):
+    subprocess.check_call("calkit list environments", shell=True)
+    # TODO: Create some environments
+
+
+def test_list_templates():
+    subprocess.check_call("calkit list templates", shell=True)
+
+
+def test_list_stages(tmp_dir):
+    subprocess.check_call("calkit list stages", shell=True)
+    ck_info = {
+        "pipeline": {
+            "stages": {
+                "stage1": {"kind": "python-script", "script_path": "train.py"},
+                "stage2": {"kind": "shell-command", "command": "echo Hello"},
+            }
+        }
+    }
+    with open("calkit.yaml", "w") as f:
+        calkit.ryaml.dump(ck_info, f)
+    out = subprocess.check_output("calkit list stages", shell=True, text=True)
+    assert "stage1" in out
+    assert "stage2" in out
+    out = subprocess.check_output(
+        "calkit list stages --kind python-script", shell=True, text=True
+    )
+    assert "stage1" in out
+    assert "stage2" not in out

calkit_python-0.34.5/calkit/tests/cli/test_list.py

@@ -1,12 +0,0 @@
-"""Tests for ``cli.list``."""
-
-import subprocess
-
-
-def test_list_environments(tmp_dir):
-    subprocess.check_call("calkit list environments", shell=True)
-    # TODO: Create some environments
-
-
-def test_list_templates():
-    subprocess.check_call("calkit list templates", shell=True)