Skip to content

Commit 1147299

Browse files
fix: librispeech waveform dtype
1 parent b2f3eaf commit 1147299

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

audio_data_pytorch/datasets/libri_speech_dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def __getitem__(
3030
) -> Union[Tensor, Tuple[Tensor, Dict]]:
3131
idx = idx.tolist() if torch.is_tensor(idx) else idx # type: ignore
3232
data = self.dataset[idx]
33-
waveform = torch.tensor(data["audio"]["array"]).view(1, -1)
33+
waveform = torch.tensor(data["audio"]["array"]).view(1, -1).float()
3434
info = dict(
3535
sample_rate=data["audio"]["sampling_rate"],
3636
text=data["text"],

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
setup(
44
name="audio-data-pytorch",
55
packages=find_packages(exclude=[]),
6-
version="0.0.11",
6+
version="0.0.12",
77
license="MIT",
88
description="Audio Data - PyTorch",
99
long_description_content_type="text/markdown",

0 commit comments

Comments
 (0)