diff --git a/environment.yml b/environment.yml
index 7cadd5cb..67f5f2fe 100644
--- a/environment.yml
+++ b/environment.yml
@@ -12,7 +12,6 @@ dependencies:
   - python=3.9.13
   - pip=23.3.1
   - pandas=2.1.4
-  - pydantic=2.0.3
   - pyg=2.5.0
   - pytorch=2.2.1
   - requests=2.31.0
@@ -23,5 +22,6 @@ dependencies:
   - torchvision=0.17.1
   - pip:
     - cellxgene-census==1.10.2 
+    - pydantic==2.6.3
     - rdkit==2023.9.5
     - yapf==0.40.2