tfdv.get_domain

tfdv.get_domain(
    schema,
    feature_name
)

Get the domain associated with the input feature from the schema.

Args:

• schema: A Schema protocol buffer.
• feature_name: The name of the feature whose domain needs to be found.

Returns:

The domain protocol buffer (one of IntDomain, FloatDomain, StringDomain or BoolDomain) associated with the input feature.

Raises:

• TypeError: If the input schema is not of the expected type.
• ValueError: If the input feature is not found in the schema or there is no domain associated with the feature.