diff --git a/src/options/generate/generator.py b/src/options/generate/generator.py index 04541cb..7860a48 100644 --- a/src/options/generate/generator.py +++ b/src/options/generate/generator.py @@ -519,3 +519,4 @@ Test scenario: def replace_with_gpt_3_5_turbo_if_possible(pkg): if pkg in ['nltk', 'textblob', 'spacy', 'transformers']: return 'gpt_3_5_turbo_api' + return pkg