Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def available_vocab_sizes(lang: str) -> Set[int]:
"""
Return the available vocabulary sizes for the given language.
Parameters
----------
lang: ``str'', required
The language identifier.
Returns
-------
The available vocabulary sizes.
"""
from .available_vocab_sizes import vocab_sizes
lang = BPEmb._get_lang(lang)
return vocab_sizes[lang]