human
I wrote the below code

    def tokenize(words):
        if isinstance(words, basestring):
            return word_tokenize(words, include_punc=False)
        else:
            return words
    all_words = chain.from_iterable((tokenize(words) for (words, _) in dataset))
    return set(all_words)

What's a good function header?
 ... Here's a good function header?
assistant
#Here's a good function header.

def _get_words_from_dataset(dataset):