azalea/CLAP
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[-] Remove deprecated torch._six

Browse Source
This commit is contained in:
Azalea Gui
2023-10-11 18:26:10 -04:00
parent 83a344004b
commit 583928827d
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/CLAPWrapper.py
+1 -2
View File
@@ -2,7 +2,6 @@ import warnings
warnings.filterwarnings("ignore") warnings.filterwarnings("ignore")
import random import random
import torchaudio import torchaudio
from torch._six import string_classes
import collections import collections
import re import re
import numpy as np import numpy as np
@@ -184,7 +183,7 @@ class CLAPWrapper():
return torch.tensor(batch, dtype=torch.float64) return torch.tensor(batch, dtype=torch.float64)
elif isinstance(elem, int): elif isinstance(elem, int):
return torch.tensor(batch) return torch.tensor(batch)
elif isinstance(elem, string_classes): elif isinstance(elem, str):
return batch return batch
elif isinstance(elem, collections.abc.Mapping): elif isinstance(elem, collections.abc.Mapping):
return {key: self.default_collate([d[key] for d in batch]) for key in elem} return {key: self.default_collate([d[key] for d in batch]) for key in elem}