lint

speechmatics · Apr 29, 2024 · a92c9f8 · a92c9f8
1 parent 2eb7f6a
commit a92c9f8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/speechmatics/cli.py b/speechmatics/cli.py
@@ -237,7 +237,9 @@ def get_transcription_config(
 
     if args.get("remove_disfluencies") is not None:
         config["transcript_filtering_config"] = {}
-        config["transcript_filtering_config"]["remove_disfluencies"] = args.get("remove_disfluencies")
+        config["transcript_filtering_config"]["remove_disfluencies"] = args.get(
+            "remove_disfluencies"
+        )
 
     if args.get("ctrl"):
         LOGGER.warning(f"Using internal dev control command: {args['ctrl']}")

diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -661,7 +661,12 @@ def test_rt_main_with_all_options(mock_server, tmp_path):
     assert msg["transcription_config"]["max_delay_mode"] == "fixed"
     assert msg["transcription_config"]["speaker_change_sensitivity"] == 0.8
     assert msg["transcription_config"].get("operating_point") is None
-    assert msg["transcription_config"]["transcript_filtering_config"]["remove_disfluencies"] is True
+    assert (
+        msg["transcription_config"]["transcript_filtering_config"][
+            "remove_disfluencies"
+        ]
+        is True
+    )
 
     # Check that the chunk size argument is respected
     add_audio_messages = mock_server.find_add_audio_messages()