nemo_automodel.components.config._arg_parser#

Module Contents#

Functions#

parse_cli_argv

Parses CLI args, pulls out –config and collects other –dotted.path options.

parse_args_and_load_config

Loads YAML, applies overrides via ConfigNode.set_by_dotted.

API#

nemo_automodel.components.config._arg_parser.parse_cli_argv(
default_cfg_path: str | None = None,
argv: list[str] | None = None,
) → tuple[str, list[str]][source]#

Parses CLI args, pulls out –config and collects other –dotted.path options.

Parameters:

  • default_cfg_path (str, optional) – Default config (yaml) path. Defaults to None.

  • argv (list[str], optional) – Argument list to parse. Defaults to sys.argv[1:].

Raises:

  • ValueError – if there’s no –config and cfg_path = None

  • ValueError – if there’s –config but not yaml file passed

Returns:

the config path along with the config overrides.

Return type:

(str, list[str])

nemo_automodel.components.config._arg_parser.parse_args_and_load_config(
default_cfg_path: str | None = None,
argv: list[str] | None = None,
) → nemo_automodel.components.config.loader.ConfigNode[source]#

Loads YAML, applies overrides via ConfigNode.set_by_dotted.

Parameters:

  • default_cfg_path (str, optional) – Default config (yaml) path.

  • argv (list[str], optional) – Argument list to parse. Defaults to sys.argv[1:].