Skip to content

Commit 1198d7e

Browse files
committed
Fixed bug where _word_set was based on train_set, even if train_set
is filelike instead of iterable.
1 parent 55e848a commit 1198d7e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

textblob/classifiers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ def __init__(self, train_set, feature_extractor=basic_extractor, format=None, **
136136
self.train_set = self._read_data(train_set, format)
137137
else: # train_set is a list of tuples
138138
self.train_set = train_set
139-
self._word_set = _get_words_from_dataset(train_set) #Keep a hidden set of unique words.
139+
self._word_set = _get_words_from_dataset(self.train_set) #Keep a hidden set of unique words.
140140
self.train_features = None
141141

142142
def _read_data(self, dataset, format=None):

0 commit comments

Comments
 (0)