diff --git a/environment.yml b/environment.yml
index ae664a542b..16d411c838 100644
--- a/environment.yml
+++ b/environment.yml
@@ -11,6 +11,7 @@ dependencies:
   - pytorch-cuda=11.7
   - pip:
     - albumentations==0.4.3
+    - antlr4-python3-runtime
     - opencv-python==4.5.5.64
     - pudb==2019.2
     - imageio==2.9.0