in retail/recommendation-system/bqml-scann/index_builder/builder/indexer.py [0:0]
def build(embedding_files_pattern, output_dir, num_leaves=None):
print("Indexer started...")
tokens, embeddings = load_embeddings(embedding_files_pattern)
index = build_index(embeddings, num_leaves)
save_index(index, tokens, output_dir)
print("Indexer finished.")