pin datasets hf package

Author: rosbo

Dockerfile.tmpl

@@ -497,7 +497,8 @@ RUN pip install flashtext && \
     pip install bqplot && \
     pip install earthengine-api && \
     pip install transformers && \
-    pip install datasets && \
+    # b/232247930 >= 2.2.0 requires pyarrow >= 6.0.0 which conflicts with dependencies for rapidsai 0.21.*
+    pip install datasets==2.1.0 && \
     pip install dlib && \
     pip install kaggle-environments && \
     pip install geopandas && \