Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions nvflow/cli/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -165,10 +165,10 @@ def list_stages(
None, "--config", "-c", help="Optional: Show only stages defined in this config file"
),
recipe: str | None = typer.Option(
None, "--recipe", "-r", help="Optional: Filter by recipe (finance, retail, healthcare)"
None, "--recipe", "-r", help="Optional: Filter by recipe (finance, example, multimodal)"
),
workflow: str | None = typer.Option(
None, "--workflow", "-w", help="Optional: Filter by workflow (training_sft, sdg_basic)"
None, "--workflow", "-w", help="Optional: Filter by workflow (training_sft, sdg_simple)"
),
):
"""List all available stages or stages in a workflow config."""
Expand Down
4 changes: 2 additions & 2 deletions nvflow/core/stage_registry.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ class StageRegistry:
"""Hierarchical registry for workflow stages.

Stages are organized in a three-level hierarchy:
- Recipe (e.g., "finance", "retail")
- Recipe (e.g., "finance", "multimodal")
- Workflow (e.g., "training_sft", "sdg_basic")
- Stage (e.g., "baseline", "generate_qa")

Expand Down Expand Up @@ -70,7 +70,7 @@ def register(cls, recipe: str, workflow: str, stage: str):
"""Decorator to register a stage class in the hierarchy.

Args:
recipe: Recipe name (e.g., "finance", "retail")
recipe: Recipe name (e.g., "finance", "multimodal")
workflow: Workflow name within recipe (e.g., "training_sft", "sdg_basic")
stage: Stage name within workflow (e.g., "baseline", "generate_qa")

Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ name = "nvflow"
dynamic = ["version"] # Version determined from git tags via hatch-vcs
description = "Workflow orchestration for end-to-end model training on NeMo ecosystem"
authors = [
{name = "Your Team", email = "team@example.com"}
{name = "NVIDIA Corporation"}
]
readme = "README.md"
requires-python = ">=3.12,<3.14"
Expand Down