forgot to commit latest version of setup.py

hipe-eval · Apr 10, 2024 · 5ededfe · 5ededfe
1 parent 54c7d83
commit 5ededfe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -24,8 +24,8 @@ def read(fname):
         "stringdist"
     ],
     extras_require = {
-        'HF': '',
-        'torch': ''
+        'huggingface': 'datasets',
+        'torch': 'torch'
     }
     long_description=read("README.md"),
     long_description_content_type='text/markdown'