Support LargeList in require_decoding

huggingface · Aug 5, 2024 · 9ec883b · 9ec883b
1 parent b27a8a1
commit 9ec883b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/datasets/features/features.py b/src/datasets/features/features.py
@@ -1595,6 +1595,8 @@ def require_decoding(feature: FeatureType, ignore_decode_attribute: bool = False
         return any(require_decoding(f) for f in feature.values())
     elif isinstance(feature, (list, tuple)):
         return require_decoding(feature[0])
+    elif isinstance(feature, LargeList):
+        return require_decoding(feature.dtype)
     elif isinstance(feature, Sequence):
         return require_decoding(feature.feature)
     else: