Add cli command (#783)

mlcommons · Jan 9, 2025 · b9aada4 · b9aada4
1 parent f71d698
commit b9aada4
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/modelgauge/main.py b/src/modelgauge/main.py
@@ -100,6 +100,15 @@ def list_suts():
         _display_factory_entry(sut_uid, sut, secrets)
 
 
+@modelgauge_cli.command()
+@LOCAL_PLUGIN_DIR_OPTION
+def list_annotators():
+    """List details about all registered SUTs (System Under Test)."""
+    secrets = load_secrets_from_config()
+    for annotator_uid, annotator in ANNOTATORS.items():
+        _display_factory_entry(annotator_uid, annotator, secrets)
+
+
 @modelgauge_cli.command()
 @LOCAL_PLUGIN_DIR_OPTION
 def list_secrets() -> None: