From 6acfef8b45240cb4e380abfe7f59729ad6aefb92 Mon Sep 17 00:00:00 2001
From: Tianyu Gao <gaotianyu1350@126.com>
Date: Tue, 11 May 2021 10:04:50 -0400
Subject: [PATCH] update setup

---
 setup.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index 1618c7cb..f91ca1d9 100644
--- a/setup.py
+++ b/setup.py
@@ -7,20 +7,21 @@
 setup(
     name='simcse',
     packages=['simcse'],
-    version='0.1',
+    version='0.2',
     license='MIT',
     description='A sentence embedding tool based on SimCSE',
     author='Tianyu Gao, Xingcheng Yao, Danqi Chen',
     author_email='tianyug@cs.princeton.edu',
     url='https://github.com/princeton-nlp/SimCSE',
-    download_url='https://github.com/princeton-nlp/SimCSE/archive/refs/tags/0.1.tar.gz',
+    download_url='https://github.com/princeton-nlp/SimCSE/archive/refs/tags/0.2.tar.gz',
     keywords=['sentence', 'embedding', 'simcse', 'nlp'],
     install_requires=[
         "tqdm",
+        "sklearn",
         "scipy>=1.5.4,<1.6",
         "transformers",
         "torch",
         "numpy>=1.19.5,<1.20",
         "setuptools"
     ]
-)
\ No newline at end of file
+)