From 8bb60321af3d94f5eb9a5a9efe5e2e57acc2bcee Mon Sep 17 00:00:00 2001 From: mlo0352 Date: Fri, 31 Mar 2023 16:57:53 -0400 Subject: [PATCH] Update utils.py Removing reference to torch._six and string_classes (they're deprecated) --- taming/data/utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/taming/data/utils.py b/taming/data/utils.py index 2b3c3d53..069e2669 100644 --- a/taming/data/utils.py +++ b/taming/data/utils.py @@ -8,7 +8,6 @@ import numpy as np import torch from taming.data.helper_types import Annotation -from torch._six import string_classes from torch.utils.data._utils.collate import np_str_obj_array_pattern, default_collate_err_msg_format from tqdm import tqdm @@ -149,7 +148,7 @@ def custom_collate(batch): return torch.tensor(batch, dtype=torch.float64) elif isinstance(elem, int): return torch.tensor(batch) - elif isinstance(elem, string_classes): + elif isinstance(elem, (str, bytes)): return batch elif isinstance(elem, collections.abc.Mapping): return {key: custom_collate([d[key] for d in batch]) for key in elem}