From 9b02b853240c9fc2bdbf60ddef412d991a1c8c40 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Florian=20Ho=CC=88nicke?= <hoenicke.florian@gmail.com>
Date: Mon, 1 May 2023 11:04:08 +0200
Subject: [PATCH] =?UTF-8?q?=F0=9F=A7=AA=20test:=20level=202?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/options/generate/generator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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