diff --git a/pyproject.toml b/pyproject.toml
index e85c396..9a28047 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -40,3 +40,7 @@ package-dir = {"" = "src"}
 
 [tool.setuptools.packages.find]
 where = ["src"]
+
+[tool.setuptools.package-data]
+trainer = ["eval_dataset/*"]
+suinput = ["data/*"]