diff --git a/pyproject.toml b/pyproject.toml
index fc943c2..f81c856 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -21,7 +21,8 @@ dependencies = [
     "matplotlib>=3.6",
     "numpy",
     "torch",
-    "scikit-learn"
+    "scikit-learn",
+    "pandas",
 ]
 
 [project.urls]