Add openai dependency

21a1178c · Leo Gao · c59cd334 · 21a1178c
Commit 21a1178c authored Jun 10, 2021 by Leo Gao
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

setup.py setup.py +3 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -37,6 +37,8 @@ setuptools.setup(
        "pybind11==2.6.2",
        "tqdm-multiprocess==0.0.11",
        "zstandard==0.15.2",
-        "jsonlines==2.0.0"
+        "jsonlines==2.0.0",
+        "mock==4.0.3",
+        "openai==0.6.4"
    ]
 )