def from_tag_name()

in tools/yaml_parser.py [0:0]

• 15 lines of code
• 2 McCabe index (conditional complexity)

  def from_tag_name(cls: Type[AbstractYamlParserType], root_dir: str,
                    tag_name: str) -> AbstractYamlParserType:
    """Builds an YamlParser from the root dir and the tag name."""
    parser_type_from_tag_name = {
        "colab": UrlYamlParser,
        "dataset": EnumerableYamlParser,
        "demo": UrlYamlParser,
        "interactive-visualizer": EnumerableYamlParser,
        "language": EnumerableYamlParser,
        "license": EnumerableYamlParser,
        "network-architecture": EnumerableYamlParser,
        "task": EnumerableYamlParser,
    }
    if tag_name not in parser_type_from_tag_name:
      raise ValueError(f"No supported parser found for tag {tag_name}.")
    return parser_type_from_tag_name[tag_name](root_dir, tag_name)