diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
index ae26c1e6..8fc58b72 100644
--- a/Dockerfile.tmpl
+++ b/Dockerfile.tmpl
@@ -557,8 +557,7 @@ RUN pip install pytorch-ignite \
         bqplot \
         earthengine-api \
         transformers \
-        # b/232247930 >= 2.2.0 requires pyarrow >= 6.0.0 which conflicts with dependencies for rapidsai 0.21.*
-        datasets==2.1.0 \
+        datasets \
         kaggle-environments \
         geopandas \
         "shapely<2" \