Update pyproject.toml

gagan3012 · Jun 15, 2021 · e8a1291 · e8a1291
1 parent e3b4518
commit e8a1291
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -21,6 +21,7 @@ requires = [
     "sentencepiece",
     "wandb",
     "pytorch_lightning",
+    "sklearn",
 ]
 license = "MIT"