black with 120 char line-length

Fix #118
rwth-i6 · Feb 28, 2023 · bcaf65b · bcaf65b
1 parent f82d114
commit bcaf65b
Show file tree

Hide file tree

Showing 47 changed files with 406 additions and 910 deletions.
diff --git a/.github/workflows/black.yml b/.github/workflows/black.yml
@@ -15,5 +15,5 @@ jobs:
       with:
         python-version: 3.8
     - run: pip install black==22.3.0
-    - run: black --diff common/
-    - run: black --check common/
+    - run: black --diff .
+    - run: black --check .
diff --git a/common/baselines/librispeech/data.py b/common/baselines/librispeech/data.py
@@ -39,9 +39,7 @@ def get_corpus_data_inputs(
     """
 
     # Dictionary containing all LibriSpeech CorpusObject entries
-    corpus_object_dict = get_corpus_object_dict(
-        audio_format="wav", output_prefix="corpora"
-    )
+    corpus_object_dict = get_corpus_object_dict(audio_format="wav", output_prefix="corpora")
 
     # Definition of the official 4-gram LM to be used as default LM
     lm = {

diff --git a/common/baselines/librispeech/default_tools.py b/common/baselines/librispeech/default_tools.py
@@ -16,9 +16,7 @@
 # RASR_BINARY_PATH = None
 # RASR_BINARY_PATH = compile_rasr_binaries_i6mode(commit="907eec4f4e36c11153f6ab6b5dd7675116f909f6")  # use tested RASR
 RASR_BINARY_PATH = compile_rasr_binaries_i6mode()  #  use most recent RASR
-assert (
-    RASR_BINARY_PATH
-), "Please set a specific RASR_BINARY_PATH before running the pipeline"
+assert RASR_BINARY_PATH, "Please set a specific RASR_BINARY_PATH before running the pipeline"
 RASR_BINARY_PATH.hash_overwrite = "LIBRISPEECH_DEFAULT_RASR_BINARY_PATH"
 
 

diff --git a/common/baselines/librispeech/ls100/gmm/baseline_args.py b/common/baselines/librispeech/ls100/gmm/baseline_args.py
@@ -139,9 +139,7 @@ def get_monophone_args():
         "use_gpu": False,
     }
 
-    return util.GmmMonophoneArgs(
-        linear_alignment_args, monophone_training_args, monophone_recognition_args
-    )
+    return util.GmmMonophoneArgs(linear_alignment_args, monophone_training_args, monophone_recognition_args)
 
 
 def get_cart_args(
@@ -163,9 +161,7 @@ def get_cart_args(
     :return:
     """
 
-    CartQuestions = (
-        CartQuestionsWithStress if use_stress_marker else CartQuestionsWithoutStress
-    )
+    CartQuestions = CartQuestionsWithStress if use_stress_marker else CartQuestionsWithoutStress
 
     cart_questions_class = CartQuestions(
         max_leaves=max_leaves,

diff --git a/common/baselines/librispeech/ls100/gmm/baseline_config.py b/common/baselines/librispeech/ls100/gmm/baseline_config.py
@@ -45,9 +45,7 @@ def run_librispeech_100_common_baseline(
     steps.add_step("vtln+sat", vtln_sat_args)
     steps.add_step("output", final_output_args)
 
-    corpus_data = get_corpus_data_inputs(
-        corpus_key="train-clean-100", use_g2p_training=True, use_stress_marker=False
-    )
+    corpus_data = get_corpus_data_inputs(corpus_key="train-clean-100", use_g2p_training=True, use_stress_marker=False)
 
     system = gmm_system.GmmSystem(rasr_binary_path=RASR_BINARY_PATH)
     system.init_system(

diff --git a/common/baselines/librispeech/ls960/gmm/baseline_args.py b/common/baselines/librispeech/ls960/gmm/baseline_args.py
@@ -143,9 +143,7 @@ def get_monophone_args(feature_flow: str = "mfcc+deriv+norm") -> util.GmmMonopho
         "use_gpu": False,
     }
 
-    return util.GmmMonophoneArgs(
-        linear_alignment_args, monophone_training_args, monophone_recognition_args
-    )
+    return util.GmmMonophoneArgs(linear_alignment_args, monophone_training_args, monophone_recognition_args)
 
 
 def get_cart_args(
@@ -168,9 +166,7 @@ def get_cart_args(
     :param add_unknown: set to true if an unknown phoneme exists
     """
 
-    CartQuestions = (
-        CartQuestionsWithStress if use_stress_marker else CartQuestionsWithoutStress
-    )
+    CartQuestions = CartQuestionsWithStress if use_stress_marker else CartQuestionsWithoutStress
 
     cart_questions_class = CartQuestions(
         max_leaves=max_leaves,

diff --git a/common/baselines/librispeech/ls960/gmm/baseline_config.py b/common/baselines/librispeech/ls960/gmm/baseline_config.py
@@ -53,9 +53,7 @@ def run_librispeech_960_common_baseline(
 
     # ******************** Data ********************
 
-    corpus_data = get_corpus_data_inputs(
-        corpus_key="train-other-960", use_g2p_training=True, use_stress_marker=False
-    )
+    corpus_data = get_corpus_data_inputs(corpus_key="train-other-960", use_g2p_training=True, use_stress_marker=False)
 
     # ******************** GMM System ********************
 

diff --git a/common/baselines/librispeech/report.py b/common/baselines/librispeech/report.py
@@ -78,14 +78,10 @@ def gmm_example_report_format(report: _Report_Type) -> str:
         + str(results["dev-other"]["VTLN"]["10"])
     )
     out.append(
-        "SAT 08".ljust(23)
-        + str(results["dev-clean"]["SAT"]["08"]).ljust(14)
-        + str(results["dev-other"]["SAT"]["08"])
+        "SAT 08".ljust(23) + str(results["dev-clean"]["SAT"]["08"]).ljust(14) + str(results["dev-other"]["SAT"]["08"])
     )
     out.append(
-        "SAT 10".ljust(23)
-        + str(results["dev-clean"]["SAT"]["10"]).ljust(14)
-        + str(results["dev-other"]["SAT"]["10"])
+        "SAT 10".ljust(23) + str(results["dev-clean"]["SAT"]["10"]).ljust(14) + str(results["dev-other"]["SAT"]["10"])
     )
     out.append(
         "VTLN+SAT 08".ljust(17)

diff --git a/common/datasets/librispeech/cart.py b/common/datasets/librispeech/cart.py
@@ -139,16 +139,12 @@ def __init__(
                     },
                     {
                         "type": "for-each-key",
-                        "keys": (" ").join(
-                            "history[0] central future[0]".split(" ")[:n_phones]
-                        ),
+                        "keys": (" ").join("history[0] central future[0]".split(" ")[:n_phones]),
                         "questions": [
                             {
                                 "type": "for-each-value",
                                 "values": self.phonemes_boundary_str,
-                                "questions": [
-                                    {"type": "question", "description": "context-phone"}
-                                ],
+                                "questions": [{"type": "question", "description": "context-phone"}],
                             },
                             {
                                 "type": "question",
@@ -374,9 +370,7 @@ def __init__(
 
 
 class CartQuestionsWithStress:
-    def __init__(
-        self, max_leaves=12001, min_obs=1000, add_unknown: bool = True, n_phones=3
-    ):
+    def __init__(self, max_leaves=12001, min_obs=1000, add_unknown: bool = True, n_phones=3):
         self.max_leaves = max_leaves
         self.min_obs = min_obs
         self.boundary = "#"
@@ -528,16 +522,12 @@ def __init__(
                     },
                     {
                         "type": "for-each-key",
-                        "keys": (" ").join(
-                            "history[0] central future[0]".split(" ")[:n_phones]
-                        ),
+                        "keys": (" ").join("history[0] central future[0]".split(" ")[:n_phones]),
                         "questions": [
                             {
                                 "type": "for-each-value",
                                 "values": self.phonemes_boundary_str,
-                                "questions": [
-                                    {"type": "question", "description": "context-phone"}
-                                ],
+                                "questions": [{"type": "question", "description": "context-phone"}],
                             },
                             {
                                 "type": "question",

diff --git a/common/datasets/librispeech/constants.py b/common/datasets/librispeech/constants.py
@@ -30,23 +30,11 @@
     "test-other": 10,
 }
 
-durations["train-clean-460"] = (
-    durations["train-clean-100"] + durations["train-clean-360"]
-)
-durations["train-other-960"] = (
-    durations["train-clean-460"] + durations["train-other-500"]
-)
+durations["train-clean-460"] = durations["train-clean-100"] + durations["train-clean-360"]
+durations["train-other-960"] = durations["train-clean-460"] + durations["train-other-500"]
 
-num_segments["train-clean-460"] = (
-    num_segments["train-clean-100"] + num_segments["train-clean-360"]
-)
-num_segments["train-other-960"] = (
-    num_segments["train-clean-460"] + num_segments["train-other-500"]
-)
+num_segments["train-clean-460"] = num_segments["train-clean-100"] + num_segments["train-clean-360"]
+num_segments["train-other-960"] = num_segments["train-clean-460"] + num_segments["train-other-500"]
 
-concurrent["train-clean-460"] = (
-    concurrent["train-clean-100"] + concurrent["train-clean-360"]
-)
-concurrent["train-other-960"] = (
-    concurrent["train-clean-460"] + concurrent["train-other-500"]
-)
+concurrent["train-clean-460"] = concurrent["train-clean-100"] + concurrent["train-clean-360"]
+concurrent["train-other-960"] = concurrent["train-clean-460"] + concurrent["train-other-500"]
diff --git a/common/datasets/librispeech/corpus.py b/common/datasets/librispeech/corpus.py
@@ -39,22 +39,16 @@ def get_bliss_corpus_dict(audio_format="flac", output_prefix="datasets"):
     output_prefix = os.path.join(output_prefix, "LibriSpeech")
 
     download_metadata_job = DownloadLibriSpeechMetadataJob()
-    download_metadata_job.add_alias(
-        os.path.join(output_prefix, "download", "metadata_job")
-    )
+    download_metadata_job.add_alias(os.path.join(output_prefix, "download", "metadata_job"))
 
     def _get_corpus(corpus_name):
         download_corpus_job = DownloadLibriSpeechCorpusJob(corpus_key=corpus_name)
         create_bliss_corpus_job = LibriSpeechCreateBlissCorpusJob(
             corpus_folder=download_corpus_job.out_corpus_folder,
             speaker_metadata=download_metadata_job.out_speakers,
         )
-        download_corpus_job.add_alias(
-            os.path.join(output_prefix, "download", corpus_name)
-        )
-        create_bliss_corpus_job.add_alias(
-            os.path.join(output_prefix, "create_bliss", corpus_name)
-        )
+        download_corpus_job.add_alias(os.path.join(output_prefix, "download", corpus_name))
+        create_bliss_corpus_job.add_alias(os.path.join(output_prefix, "create_bliss", corpus_name))
         return create_bliss_corpus_job.out_corpus
 
     corpus_names = [
@@ -67,9 +61,7 @@ def _get_corpus(corpus_name):
         "train-other-500",
     ]
 
-    bliss_corpus_dict = {
-        corpus_name: _get_corpus(corpus_name) for corpus_name in corpus_names
-    }
+    bliss_corpus_dict = {corpus_name: _get_corpus(corpus_name) for corpus_name in corpus_names}
 
     audio_format_options = {
         "wav": {
@@ -94,19 +86,13 @@ def _get_corpus(corpus_name):
                     corpus_name,
                 )
             )
-            converted_bliss_corpus_dict[
-                corpus_name
-            ] = bliss_change_encoding_job.out_corpus
+            converted_bliss_corpus_dict[corpus_name] = bliss_change_encoding_job.out_corpus
     else:
         converted_bliss_corpus_dict = bliss_corpus_dict
 
     def _merge_corpora(corpora, name):
-        merge_job = MergeCorporaJob(
-            bliss_corpora=corpora, name=name, merge_strategy=MergeStrategy.FLAT
-        )
-        merge_job.add_alias(
-            os.path.join(output_prefix, "%s_merge" % audio_format, name)
-        )
+        merge_job = MergeCorporaJob(bliss_corpora=corpora, name=name, merge_strategy=MergeStrategy.FLAT)
+        merge_job.add_alias(os.path.join(output_prefix, "%s_merge" % audio_format, name))
         return merge_job.out_merged_corpus
 
     converted_bliss_corpus_dict["train-clean-460"] = _merge_corpora(
@@ -151,9 +137,7 @@ def get_corpus_object_dict(audio_format="flac", output_prefix="datasets"):
         - 'train-other-960'
     :rtype: dict[str, CorpusObject]
     """
-    bliss_corpus_dict = get_bliss_corpus_dict(
-        audio_format=audio_format, output_prefix=output_prefix
-    )
+    bliss_corpus_dict = get_bliss_corpus_dict(audio_format=audio_format, output_prefix=output_prefix)
 
     corpus_object_dict = {}
 
@@ -196,19 +180,15 @@ def get_ogg_zip_dict(
     from i6_core.returnn.oggzip import BlissToOggZipJob
 
     ogg_zip_dict = {}
-    bliss_corpus_dict = get_bliss_corpus_dict(
-        audio_format="ogg", output_prefix=output_prefix
-    )
+    bliss_corpus_dict = get_bliss_corpus_dict(audio_format="ogg", output_prefix=output_prefix)
     for name, bliss_corpus in bliss_corpus_dict.items():
         ogg_zip_job = BlissToOggZipJob(
             bliss_corpus,
             no_conversion=True,
             returnn_python_exe=returnn_python_exe,
             returnn_root=returnn_root,
         )
-        ogg_zip_job.add_alias(
-            os.path.join(output_prefix, "LibriSpeech", "%s_ogg_zip_job" % name)
-        )
+        ogg_zip_job.add_alias(os.path.join(output_prefix, "LibriSpeech", "%s_ogg_zip_job" % name))
         ogg_zip_dict[name] = ogg_zip_job.out_ogg_zip
 
     return ogg_zip_dict
diff --git a/common/datasets/librispeech/export.py b/common/datasets/librispeech/export.py
@@ -15,23 +15,17 @@ def _export_datasets(output_prefix):
 
     # export all bliss corpora
     for audio_format in ["flac", "ogg", "wav"]:
-        bliss_corpus_dict = get_bliss_corpus_dict(
-            audio_format=audio_format, output_prefix=output_prefix
-        )
+        bliss_corpus_dict = get_bliss_corpus_dict(audio_format=audio_format, output_prefix=output_prefix)
         for name, bliss_corpus in bliss_corpus_dict.items():
             tk.register_output(
-                os.path.join(
-                    output_prefix, "LibriSpeech", "%s-%s.xml.gz" % (name, audio_format)
-                ),
+                os.path.join(output_prefix, "LibriSpeech", "%s-%s.xml.gz" % (name, audio_format)),
                 bliss_corpus,
             )
 
     # export all ogg zip corpora
     ogg_corpus_dict = get_ogg_zip_dict(output_prefix=output_prefix)
     for name, ogg_corpus in ogg_corpus_dict.items():
-        tk.register_output(
-            os.path.join(output_prefix, "LibriSpeech", "%s.ogg.zip" % name), ogg_corpus
-        )
+        tk.register_output(os.path.join(output_prefix, "LibriSpeech", "%s.ogg.zip" % name), ogg_corpus)
 
 
 def _export_lm_data(output_prefix):
@@ -57,37 +51,27 @@ def _export_lexicon_and_vocab(output_prefix):
     lexicon_output_prefix = os.path.join(output_prefix, "LibriSpeech", "lexicon")
 
     # folded / without stress marker
-    bliss_lexicon = get_bliss_lexicon(
-        output_prefix=output_prefix, use_stress_marker=False
-    )
+    bliss_lexicon = get_bliss_lexicon(output_prefix=output_prefix, use_stress_marker=False)
     tk.register_output(
         os.path.join(lexicon_output_prefix, "librispeech.lexicon.folded.xml.gz"),
         bliss_lexicon,
     )
 
-    g2p_lexicon_dict = get_g2p_augmented_bliss_lexicon_dict(
-        use_stress_marker=True, output_prefix=output_prefix
-    )
+    g2p_lexicon_dict = get_g2p_augmented_bliss_lexicon_dict(use_stress_marker=True, output_prefix=output_prefix)
     for k, lexicon in g2p_lexicon_dict.items():
         tk.register_output(
-            os.path.join(
-                lexicon_output_prefix, "%s.lexicon_with_g2p.folded.xml.gz" % k
-            ),
+            os.path.join(lexicon_output_prefix, "%s.lexicon_with_g2p.folded.xml.gz" % k),
             lexicon,
         )
 
     # with stress marker
-    bliss_lexicon = get_bliss_lexicon(
-        output_prefix=output_prefix, use_stress_marker=True
-    )
+    bliss_lexicon = get_bliss_lexicon(output_prefix=output_prefix, use_stress_marker=True)
     tk.register_output(
         os.path.join(lexicon_output_prefix, "librispeech.lexicon.xml.gz"),
         bliss_lexicon,
     )
 
-    g2p_lexicon_dict = get_g2p_augmented_bliss_lexicon_dict(
-        use_stress_marker=False, output_prefix=output_prefix
-    )
+    g2p_lexicon_dict = get_g2p_augmented_bliss_lexicon_dict(use_stress_marker=False, output_prefix=output_prefix)
     for k, lexicon in g2p_lexicon_dict.items():
         tk.register_output(
             os.path.join(lexicon_output_prefix, "%s.lexicon_with_g2p.xml.gz" % k),
@@ -102,12 +86,8 @@ def _export_legacy_bpe(output_prefix):
     :param str output_prefix
     """
     lexicon_output_prefix = os.path.join(output_prefix, "LibriSpeech", "bpe")
-    ls960_bpe_settings = get_subword_nmt_bpe(
-        corpus_key="train-other-960", bpe_size=10000, output_prefix=output_prefix
-    )
-    ls100_bpe_settings = get_subword_nmt_bpe(
-        corpus_key="train-clean-100", bpe_size=2000, output_prefix=output_prefix
-    )
+    ls960_bpe_settings = get_subword_nmt_bpe(corpus_key="train-other-960", bpe_size=10000, output_prefix=output_prefix)
+    ls100_bpe_settings = get_subword_nmt_bpe(corpus_key="train-clean-100", bpe_size=2000, output_prefix=output_prefix)
     tk.register_output(
         os.path.join(lexicon_output_prefix, "train-other-960", "bpe_10k.codes"),
         ls960_bpe_settings.bpe_codes,

diff --git a/common/datasets/librispeech/language_model.py b/common/datasets/librispeech/language_model.py
@@ -33,12 +33,8 @@ def get_arpa_lm_dict(output_prefix="datasets"):
     lm_dict["3gram"] = download_arpa_3gram_lm_job.out_file
 
     lm_prefix = os.path.join(output_prefix, "LibriSpeech", "lm")
-    download_arpa_3gram_lm_job.add_alias(
-        os.path.join(lm_prefix, "download_3gram_lm_job")
-    )
-    download_arpa_4gram_lm_job.add_alias(
-        os.path.join(lm_prefix, "download_4gram_lm_job")
-    )
+    download_arpa_3gram_lm_job.add_alias(os.path.join(lm_prefix, "download_3gram_lm_job"))
+    download_arpa_4gram_lm_job.add_alias(os.path.join(lm_prefix, "download_4gram_lm_job"))
 
     return lm_dict
 
@@ -51,10 +47,6 @@ def get_librispeech_normalized_lm_data(output_prefix="datasets") -> tk.Path:
     :param output_prefix:
     :return: gzipped text file containing the LM training data
     """
-    download_job = DownloadJob(
-        url="https://www.openslr.org/resources/11/librispeech-lm-norm.txt.gz"
-    )
-    download_job.add_alias(
-        os.path.join(output_prefix, "LibriSpeech", "lm", "download_lm_data")
-    )
+    download_job = DownloadJob(url="https://www.openslr.org/resources/11/librispeech-lm-norm.txt.gz")
+    download_job.add_alias(os.path.join(output_prefix, "LibriSpeech", "lm", "download_lm_data"))
     return download_job.out_file