Skip to content

Commit

Permalink
Add docstrings
Browse files Browse the repository at this point in the history
  • Loading branch information
dagardner-nv committed Jan 23, 2023
1 parent 9b39443 commit a94783c
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions morpheus/cli/default_command_hooks.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,17 +23,24 @@


class DefaultCommandHooks:
"""
Hooks for registering and collecting stages to execute as part of a morhpeus pipeline
"""

@hookimpl
def morpheus_cli_collect_stage_names(self, mode: PipelineModes) -> typing.List[str]:

# Loop over the existing stage registry and return the names
"""
Loop over the existing stage registry and return the names
"""
command_names = StageRegistry.get_registered_names(mode=mode)

return command_names

@hookimpl
def morpheus_cli_make_stage_command(self, mode: PipelineModes, stage_name: str) -> click.Command:
"""
Add a stage command to a pipeline mode based on info in the stage registry
"""

stage_info = StageRegistry.get_stage_info(stage_name=stage_name, mode=mode, raise_missing=False)

Expand Down

0 comments on commit a94783c

Please sign in to comment.