espnet3.components.data.dataset_module.load_dataset_module

source

espnet3.components.data.dataset_module.load_dataset_module(data_src: str | None = None, recipe_dir: str | Path | None = None)

Load a dataset module from a dataset source or local recipe path.

• Parameters:
  • data_src – Dataset source reference. Supported forms are listed below.
    • None: load local recipe_dir/dataset module.
    • tag form: mini_an4/asr.
    • module path form: egs3.mini_an4.asr.dataset.
  • recipe_dir – Recipe root directory used for local module loading.
• Returns: Imported dataset module object.
• Raises:
  • AssertionError – If both data_src and recipe_dir are None.
  • ModuleNotFoundError – If target module cannot be found.
  • ImportError – If import fails for other reasons.

Notes

This function does not instantiate Dataset or DatasetBuilder. It only resolves and imports the module.

Examples

>>> module = load_dataset_module(data_src="mini_an4/asr")
>>> hasattr(module, "DatasetBuilder")
True
>>> local = load_dataset_module(
...     data_src=None, recipe_dir="egs3/mini_an4/asr"
... )
>>> hasattr(local, "Dataset")
True
>>> load_dataset_module(data_src=None, recipe_dir=None)
Traceback (most recent call last):
...
AssertionError: recipe_dir must be set when data_src is None.