diff --git a/pl_bolts/datamodules/async_dataloader.py b/pl_bolts/datamodules/async_dataloader.py index 93ddc84add..0767ad4a1a 100644 --- a/pl_bolts/datamodules/async_dataloader.py +++ b/pl_bolts/datamodules/async_dataloader.py @@ -6,7 +6,6 @@ import torch from torch import Tensor -from torch._six import string_classes from torch.utils.data import DataLoader, Dataset from pl_bolts.utils.stability import under_review @@ -85,7 +84,7 @@ def load_instance(self, sample: Any) -> Any: return {key: self.load_instance(sample[key]) for key in sample} elif isinstance(sample, tuple) and hasattr(sample, "_fields"): # namedtuple return elem_type(*(self.load_instance(d) for d in sample)) - elif isinstance(sample, container_abcs.Sequence) and not isinstance(sample, string_classes): + elif isinstance(sample, container_abcs.Sequence) and not isinstance(sample, str): return [self.load_instance(s) for s in sample] else: return sample