from sklearn import TfIdfVectorizer
import wordcloud
from pydantic import BaseModel


class WordCloudExtractor:

    tfidf_params: Dict[str, Any]

    def extract_from_corpus(self, texts: Iterable[str], n_words: int) -> wordcloud.WordCloud:
        pass