Update tft.py

AnandInguva · web-flow · commit 6472ba7b3968 · 2023-10-24T15:38:27.000-04:00
diff --git a/sdks/python/apache_beam/ml/transforms/tft.py b/sdks/python/apache_beam/ml/transforms/tft.py
@@ -558,7 +558,7 @@ def __init__(
     if compute_word_count:
       self.compute_word_count_fn = count_unqiue_words
     else:
-      self.compute_word_count_fn = lambda *args, **kwargs: {}
+      self.compute_word_count_fn = lambda *args, **kwargs: None
 
     if ngram_range != (1, 1) and not ngrams_separator:
       raise ValueError(
