Commit 5d6cd874 authored by Phil Wang's avatar Phil Wang
Browse files

reduce confusion for target_sample_hz

parent 0f10c2e1

audiolm_pytorch/data.py

+4 −2
Original line number Diff line number Diff line 
from pathlib import Path

from functools import partial, wraps



from beartype.typing import Tuple

from beartype import beartype

from beartype.typing import Tuple, Union, Optional

from beartype.door import is_bearable



import torchaudio
@@ -25,13 +26,14 @@ def cast_tuple(val, length = 1): 


# dataset functions



@beartype

class SoundDataset(Dataset):

    def __init__(

        self,

        folder,

        exts = ['flac', 'wav'],

        max_length = None,

        target_sample_hz = None,

        target_sample_hz: Optional[Tuple[Optional[int], ...]] = None,

        seq_len_multiple_of = None

    ):

        super().__init__()

setup.py

+1 −1
Original line number Diff line number Diff line
@@ -3,7 +3,7 @@ from setuptools import setup, find_packages 
setup(

  name = 'audiolm-pytorch',

  packages = find_packages(exclude=[]),

  version = '0.4.2',

  version = '0.4.3',

  license='MIT',

  description = 'AudioLM - Language Modeling Approach to Audio Generation from Google Research - Pytorch',

  author = 'Phil Wang',