Fixes

Lightning-Universe · Jan 5, 2022 · b9077cc · b9077cc
1 parent 65a37b2
commit b9077cc
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/flash/audio/speech_recognition/cli.py b/flash/audio/speech_recognition/cli.py
@@ -28,8 +28,8 @@ def from_timit(
     """Downloads and loads the timit data set."""
     download_data("https://pl-flash-data.s3.amazonaws.com/timit_data.zip", "./data")
     return SpeechRecognitionData.from_json(
-        input_fields="file",
-        target_fields="text",
+        "file",
+        "text",
         train_file="data/timit/train.json",
         test_file="data/timit/test.json",
         val_split=val_split,

diff --git a/flash/audio/speech_recognition/data.py b/flash/audio/speech_recognition/data.py
@@ -423,6 +423,7 @@ def from_datasets(
         objects.
 
         The Dataset objects should be one of the following:
+
         * A PyTorch Dataset where the ``__getitem__`` returns a tuple: ``(file_path or , target)``
         * A PyTorch Dataset where the ``__getitem__`` returns a dict: ``{"input": file_path, "target": target}``
 

diff --git a/flash_examples/speech_recognition.py b/flash_examples/speech_recognition.py
@@ -21,8 +21,8 @@
 download_data("https://pl-flash-data.s3.amazonaws.com/timit_data.zip", "./data")
 
 datamodule = SpeechRecognitionData.from_json(
-    input_fields="file",
-    target_fields="text",
+    "file",
+    "text",
     train_file="data/timit/train.json",
     test_file="data/timit/test.json",
     batch_size=4,