From c8493378c9e8db74efcd30f96247923069cf53bd Mon Sep 17 00:00:00 2001
From: Philmod <philmod@google.com>
Date: Fri, 22 Dec 2023 10:38:16 +0100
Subject: [PATCH] Unpin datasets

---
 Dockerfile.tmpl | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

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" \