diff --git a/video_prediction_tools/env_setup/requirements.txt b/video_prediction_tools/env_setup/requirements.txt
index d3d609fdec3ef7e5de406e21812cba59d4df1d25..a831d7d8daf0159ad3e1c5709db9bb9a71cf5779 100755
--- a/video_prediction_tools/env_setup/requirements.txt
+++ b/video_prediction_tools/env_setup/requirements.txt
@@ -1,5 +1,7 @@
 matplotlib==3.3.0
-mpi4py==3.0.3
+#mpi4py==3.0.3
 pandas==0.25.3
+xarray==0.16.0
+basemap==1.3.0
 scikit-image==0.18.1
 opencv-python-headless==4.2.0.34