update setup

Author: Tianyu Gao

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='[email protected]',
     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"
     ]
-)
+)