Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

remove reference to torch._six #993

Merged
merged 8 commits into from
Mar 28, 2023
+1 −2
Merged

remove reference to torch._six #993

Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
01b5fed
remove reference to torch._six
cdeepali Mar 17, 2023
14dbdd7
Merge branch 'master' into dc-remove-torchsix
Borda Mar 28, 2023
27b94d8
req.
Borda Mar 28, 2023
1e50021
freeze req.
Borda Mar 28, 2023
7255548
Merge branch 'master' into dc-remove-torchsix
Borda Mar 28, 2023
e80f685
freeze req.
Borda Mar 28, 2023
807485f
Merge branch 'master' into dc-remove-torchsix
Borda Mar 28, 2023
505a0b8
Merge branch 'master' into dc-remove-torchsix
Borda Mar 28, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions pl_bolts/datamodules/async_dataloader.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down