From 5ededfed8cc20b58e30de728f65cfd9e8279ed47 Mon Sep 17 00:00:00 2001
From: Matteo Romanello <matteo.romanello@epfl.ch>
Date: Wed, 10 Apr 2024 11:23:45 +0200
Subject: [PATCH] forgot to commit latest version of setup.py

---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index b9dd776..207fc50 100644
--- 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'