Add promptsource to setup.py

ec2eb117 · cjlovering · c93093b6 · ec2eb117
Commit ec2eb117 authored Apr 25, 2022 by cjlovering
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

setup.py setup.py +4 -3

No files found.
--- a/setup.py
+++ b/setup.py
@@ -18,8 +18,9 @@ setuptools.setup(
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
    ],
-    python_requires='>=3.6',
+    python_requires=">=3.6",
    install_requires=[
+        "promptsource",
        "black",
        "datasets==2.0.0",
        "click>=7.1",
@@ -42,9 +43,9 @@ setuptools.setup(
        "openai==0.6.4",
        "jieba==0.42.1",
        "nagisa==0.2.7",
-        "bleurt@https://github.com/google-research/bleurt/archive/b610120347ef22b494b6d69b4316e303f5932516.zip#egg=bleurt"
+        "bleurt@https://github.com/google-research/bleurt/archive/b610120347ef22b494b6d69b4316e303f5932516.zip#egg=bleurt",
    ],
    dependency_links=[
        "https://github.com/google-research/bleurt/archive/b610120347ef22b494b6d69b4316e303f5932516.zip#egg=bleurt",
-    ]
+    ],
 )