fix: remove lambda

archinetai · Oct 11, 2022 · 1d6e59c · 1d6e59c
1 parent 17b2786
commit 1d6e59c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/audio_data_pytorch/datasets/audio_web_dataset.py b/audio_data_pytorch/datasets/audio_web_dataset.py
@@ -157,6 +157,10 @@ def identity(x):
     return x
 
 
+def first(x):
+    return x[0]
+
+
 class AudioWebDataset(WebDataset):
 
     # Why batch_size in a dataset constructor?
@@ -176,7 +180,7 @@ def __init__(
             self.shuffle(shuffle)
             .decode(torch_audio)
             .to_tuple("wav", "json")
-            .map_tuple((lambda tuple: tuple[0]), identity)
+            .map_tuple(first, identity)
         )
 
         if exists(transforms):

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
     name="audio-data-pytorch",
     packages=find_packages(exclude=[]),
-    version="0.0.14",
+    version="0.0.15",
     license="MIT",
     description="Audio Data - PyTorch",
     long_description_content_type="text/markdown",