diff --git a/environment.yml b/environment.yml
index 01883453..4fc136a5 100644
--- a/environment.yml
+++ b/environment.yml
@@ -17,3 +17,4 @@ dependencies:
   - torchtnt
   - pip:
     - torchnet
+    - numpy<2