Update setup.py

Tina2m · Nov 16, 2023 · 4196e5e · 4196e5e
1 parent bbb909c
commit 4196e5e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -9,7 +9,7 @@
         'numpy',
         'pandas',
         'einops',
-        'sklearn',
+        'scikit-learn',
         'wandb',
         'jupyter',
         'matplotlib',
@@ -25,4 +25,4 @@
         'rouge_score==0.1.2',
         'gradio==3.36.1'
     ]
-)
+)