remove reference to torch._six (#993)

Lightning-Universe · Mar 28, 2023 · c4e717c · c4e717c
1 parent 74f3369
commit c4e717c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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