From f5bd2a474fcdd61484b275141d218c0a33697596 Mon Sep 17 00:00:00 2001 From: Guillaume Lagrange Date: Fri, 23 Feb 2024 10:44:03 -0500 Subject: [PATCH] Check that pa_type is valid before checking if is_binary (#1354) --- crates/burn-dataset/src/source/huggingface/importer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/crates/burn-dataset/src/source/huggingface/importer.py b/crates/burn-dataset/src/source/huggingface/importer.py index f5d3ff950..0c3079b30 100644 --- a/crates/burn-dataset/src/source/huggingface/importer.py +++ b/crates/burn-dataset/src/source/huggingface/importer.py @@ -102,7 +102,7 @@ def blob_columns(dataset): """ type_mapping = {} for name, value in dataset.features.items(): - if pa.types.is_binary(value.pa_type): + if value.pa_type is not None and pa.types.is_binary(value.pa_type): type_mapping[name] = LargeBinary return type_mapping