update 0.3

princeton-nlp · May 11, 2021 · 725f70e · 725f70e
1 parent 6acfef8
commit 725f70e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -7,17 +7,17 @@
 setup(
     name='simcse',
     packages=['simcse'],
-    version='0.2',
+    version='0.3',
     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.2.tar.gz',
+    download_url='https://github.com/princeton-nlp/SimCSE/archive/refs/tags/0.3.tar.gz',
     keywords=['sentence', 'embedding', 'simcse', 'nlp'],
     install_requires=[
         "tqdm",
-        "sklearn",
+        "scikit-learn",
         "scipy>=1.5.4,<1.6",
         "transformers",
         "torch",