Skip to content

Commit

Permalink
Refactor list_registered_models to return None and simplify docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
cpaniaguam committed Feb 8, 2025
1 parent 31e3099 commit 2109bd1
Showing 1 changed file with 2 additions and 8 deletions.
10 changes: 2 additions & 8 deletions src/hssm/register.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,14 +60,8 @@ def register_model(
registered_models[name] = config


def list_registered_models() -> dict[SupportedModels, str]:
"""List all registered models and their descriptions.
Returns
-------
dict[SupportedModels, str]
Dictionary mapping model names to their descriptions
"""
def list_registered_models() -> None:
"""List all registered models and their descriptions."""
pp(
{
name: config.get("description") or "No description"
Expand Down

0 comments on commit 2109bd1

Please sign in to comment.