apache_beam.ml.transforms.utils module

class apache_beam.ml.transforms.utils.ArtifactsFetcher(artifact_location: str)[source]

Bases: object

Utility class used to fetch artifacts from the artifact_location passed to the TFTProcessHandlers in MLTransform.

This is intended to be used for testing purposes only.

get_vocab_list(vocab_filename: str) list[bytes][source]

Returns list of vocabulary terms created during MLTransform.

get_vocab_filepath(vocab_filename: str) str[source]

Return the path to the vocabulary file created during MLTransform.

get_vocab_size(vocab_filename: str) int[source]