diff --git a/src/options/generate/generator.py b/src/options/generate/generator.py index a9f2bc9..8ea9300 100644 --- a/src/options/generate/generator.py +++ b/src/options/generate/generator.py @@ -530,6 +530,6 @@ Test scenario: @staticmethod def replace_with_gpt_3_5_turbo_if_possible(pkg): - if pkg in ['nltk', 'textblob', 'spacy', 'transformers']: + if pkg in ['nltk', 'textblob', 'spacy', 'transformers', 'textstat']: return 'gpt_3_5_turbo_api' return pkg