diff --git a/nvflow/cli/main.py b/nvflow/cli/main.py index fbfe72e..fd57bd6 100644 --- a/nvflow/cli/main.py +++ b/nvflow/cli/main.py @@ -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.""" diff --git a/nvflow/core/stage_registry.py b/nvflow/core/stage_registry.py index 49daae9..3081539 100644 --- a/nvflow/core/stage_registry.py +++ b/nvflow/core/stage_registry.py @@ -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") @@ -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") diff --git a/pyproject.toml b/pyproject.toml index 31761ea..81ab51d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -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"