From 1c785d5523204d74919a9655f375e890948826f9 Mon Sep 17 00:00:00 2001
From: Roee Zamir <69147754+roeezamir@users.noreply.github.com>
Date: Mon, 20 Feb 2023 11:04:24 +0200
Subject: [PATCH] Update explore_data.py

Set the CountVectorizer with the correct ngram_range argument.
---
 ml/guides/text_classification/explore_data.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ml/guides/text_classification/explore_data.py b/ml/guides/text_classification/explore_data.py
index cbf454f..72f6baa 100644
--- a/ml/guides/text_classification/explore_data.py
+++ b/ml/guides/text_classification/explore_data.py
@@ -72,7 +72,7 @@ def plot_frequency_distribution_of_ngrams(sample_texts,
     """
     # Create args required for vectorizing.
     kwargs = {
-            'ngram_range': (1, 1),
+            'ngram_range':  ngram_range,
             'dtype': 'int32',
             'strip_accents': 'unicode',
             'decode_error': 'replace',