diff --git a/pyproject.toml b/pyproject.toml
index 75c6b28..40efcc3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -23,6 +23,8 @@ dependencies = [
     "packaging",
     "cached_path>=1.6.2",
     "transformers",
+    "matplotlib",
+    "scikit-learn"
 ]
 
 [project.optional-dependencies]