diff --git a/epos-ai-train.yml b/epos-ai-train.yml
index 0ecd808..a643060 100644
--- a/epos-ai-train.yml
+++ b/epos-ai-train.yml
@@ -15,6 +15,7 @@ dependencies:
   - pytorch-lightning
   - scikit-learn
   - openpyxl
+  - pydantic
   - jupyterlab
   - notebook
   - pip