diff --git a/tensorflow_gnn/converters/ogb/ogb_lib.py b/tensorflow_gnn/converters/ogb/ogb_lib.py index 3d92e459..132bd301 100644 --- a/tensorflow_gnn/converters/ogb/ogb_lib.py +++ b/tensorflow_gnn/converters/ogb/ogb_lib.py @@ -44,7 +44,7 @@ def generate_examples(features: DataTable, value = feat.float_list.value elif array.dtype.type in (numpy.bytes_,): value = feat.bytes_list.value - elif array.dtype.type in (numpy.string_, numpy.str_): + elif array.dtype.type in (numpy.bytes_, numpy.str_): value = [word.encode("utf-8") for word in feat.bytes_list.value] else: raise NotImplementedError(