mirror of https://github.com/tracel-ai/burn.git
Check that pa_type is valid before checking if is_binary (#1354)
This commit is contained in:
parent
08302e38fc
commit
f5bd2a474f
|
@ -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
|
||||
|
||||
|
|
Loading…
Reference in New Issue