From 76c77cd28ae31bb3a53a8d52ea41ed3a24f81665 Mon Sep 17 00:00:00 2001 From: Jack McCluskey <34928439+jrmccluskey@users.noreply.github.com> Date: Thu, 18 Apr 2024 15:57:55 -0400 Subject: [PATCH] Fix typo in count_unique_words() (#31023) --- sdks/python/apache_beam/ml/transforms/tft.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sdks/python/apache_beam/ml/transforms/tft.py b/sdks/python/apache_beam/ml/transforms/tft.py index f1ff0e4dfe57..cf44ee9ca0fb 100644 --- a/sdks/python/apache_beam/ml/transforms/tft.py +++ b/sdks/python/apache_beam/ml/transforms/tft.py @@ -575,7 +575,7 @@ def __init__( self.split_string_by_delimiter = split_string_by_delimiter self.key_vocab_filename = key_vocab_filename if compute_word_count: - self.compute_word_count_fn = count_unqiue_words + self.compute_word_count_fn = count_unique_words else: self.compute_word_count_fn = lambda *args, **kwargs: None @@ -597,6 +597,6 @@ def apply_transform(self, data: tf.SparseTensor, output_col_name: str): return {output_col_name: output} -def count_unqiue_words( +def count_unique_words( data: tf.SparseTensor, output_vocab_name: Optional[str]) -> None: tft.count_per_key(data, key_vocabulary_filename=output_vocab_name)