[-] Remove deprecated torch._six

This commit is contained in:
2023-10-11 18:26:10 -04:00
parent 83a344004b
commit 583928827d
+1 -2
View File
@@ -2,7 +2,6 @@ import warnings
warnings.filterwarnings("ignore")
import random
import torchaudio
from torch._six import string_classes
import collections
import re
import numpy as np
@@ -184,7 +183,7 @@ class CLAPWrapper():
return torch.tensor(batch, dtype=torch.float64)
elif isinstance(elem, int):
return torch.tensor(batch)
elif isinstance(elem, string_classes):
elif isinstance(elem, str):
return batch
elif isinstance(elem, collections.abc.Mapping):
return {key: self.default_collate([d[key] for d in batch]) for key in elem}