Fix installable

2021-01-26 17:39:37 +11:00 · 2021-01-26 17:39:37 +11:00 · aba7823739
commit aba7823739
parent 50d5659d73
1 changed files with 4 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@ -5,8 +5,10 @@ import os
 from setuptools import setup, find_packages

 core_req = ['ftfy', 'regex', 'tqdm', 'torch==1.7.1', 'torchvision']
-extras_require={'cuda': ['cudatoolkit==11.0'],
-                'dev': ['pytest']}
+extras_require={
+    'cuda': ['cudatoolkit==11.0'],
+    'dev': ['pytest']
+}

 setup(
    name='clip_by_openai',