Add support for huggingface ASR models in hg recipes

src/fairseq2/recipes/hg/__init__.py

@@ -6,44 +6,40 @@
 
 from __future__ import annotations
 
-try:
-    import datasets  # type: ignore[attr-defined,import-untyped,import-not-found]
+from typing import Dict, List
 
-    _has_hg_datasets = True
-except ImportError:
-    _has_hg_datasets = False
-
-
-try:
-    import evaluate  # type: ignore[attr-defined,import-untyped,import-not-found]
-
-    _has_hg_evaluate = True
-except ImportError:
-    _has_hg_evaluate = False
+from fairseq2.recipes.cli import Cli, RecipeCommandHandler
 
 
-from fairseq2.recipes.cli import Cli, RecipeCommandHandler
+def check_libraries(libraries: List[str]) -> Dict[str, bool]:
+    """Check if the given libraries are available."""
+    availability = {}
+    for lib in libraries:
+        try:
+            __import__(lib)
+            availability[lib] = True
+        except ImportError:
+            availability[lib] = False
+    return availability
 
 
 def _setup_hg_cli(cli: Cli) -> None:
-    if not _has_hg_datasets or not _has_hg_evaluate:
+    required_libraries = ["transformers", "datasets", "evaluate", "hydra"]
+    if not all(check_libraries(required_libraries).values()):
         return
 
     group = cli.add_group("hg", help="Hugging Face recipes")
 
-    from fairseq2.recipes.hg.asr_eval import (
-        asr_eval_presets,
-        load_wav2vec2_asr_evaluator,
-    )
+    from fairseq2.recipes.hg.asr_eval import asr_eval_presets, load_asr_evaluator
 
     handler = RecipeCommandHandler(
-        load_wav2vec2_asr_evaluator,
+        load_asr_evaluator,
         preset_configs=asr_eval_presets,
-        default_preset="librispeech_asr",
+        default_preset="default_asr",
     )
 
     group.add_command(
-        "wav2vec2_asr",
+        "asr",
         handler,
-        help="evaluate a wav2vec 2.0 ASR model on a downstream benchmark (default: librispeech_asr)",
+        help="evaluate an ASR model (default: wav2vec2) on a downstream benchmark (default: librispeech_asr)",
     )