diff --git a/continuous_integration/environment-actions.yml b/continuous_integration/environment-actions.yml
index dd28266..f8e6c5d 100644
--- a/continuous_integration/environment-actions.yml
+++ b/continuous_integration/environment-actions.yml
@@ -23,3 +23,4 @@ dependencies:
   - jaxopt
   - tensorflow>=2.6
   - tensorflow-probability
+  - xarray-datatree