From 85ab9fee77efb3fbc96577d2294fa998cb8fad96 Mon Sep 17 00:00:00 2001
From: Longhu Qin <46173279+KenelmQLH@users.noreply.github.com>
Date: Mon, 4 Mar 2024 16:20:59 +0800
Subject: [PATCH 1/2] Update setup.py

---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index 7d58563c..16dbf8cf 100644
--- a/setup.py
+++ b/setup.py
@@ -38,6 +38,7 @@
 
 vec_deps = [
     'gensim',
+    'smart-open==6.4.0',
     'transformers<4.29.0',
     "tensorboard",
     'torchvision',

From d6751434eb9a8be23217a3d1c2220a7d24d7359b Mon Sep 17 00:00:00 2001
From: Longhu Qin <46173279+KenelmQLH@users.noreply.github.com>
Date: Mon, 4 Mar 2024 17:26:43 +0800
Subject: [PATCH 2/2] Update setup.py

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 16dbf8cf..c5cc9e21 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,7 @@
     "ipython"
 ]
 test_deps = [
-    'pytest>=4',
+    'pytest<=8.0.2',
     'pytest-cov>=2.6.0',
     'pytest-flake8',
     'flake8<5.0.0'