lint test changes

Author: James Walker

speechmatics/cli.py

@@ -840,7 +840,9 @@ def run(stream=None, channel_stream_pairs=None):
                 else:
                     channel_stream_pairs = {}
                     for i in range(num_channels):
-                        channel_name = transcription_config.channel_diarization_labels[i]
+                        channel_name = transcription_config.channel_diarization_labels[
+                            i
+                        ]
                         channel_stream_pairs[channel_name] = args["files"][i]
                     run(channel_stream_pairs=channel_stream_pairs)
             else:

speechmatics/client.py

@@ -265,7 +265,7 @@ async def _producer(self, stream, audio_chunk_size):
             async for audio_chunk in read_in_chunks(stream, audio_chunk_size):
                 if self._session_needs_closing:
                     break
-                
+
                 if self._transcription_config_needs_update:
                     yield self._set_recognition_config()
                     self._transcription_config_needs_update = False

tests/test_cli.py

@@ -76,10 +76,17 @@
             },
         ),
         (
-            ["rt", "transcribe", "--multichannel", "test_channel_1,test_channel_2", "diarization", "speaker"],
+            [
+                "rt",
+                "transcribe",
+                "--multichannel",
+                "test_channel_1,test_channel_2",
+                "diarization",
+                "speaker",
+            ],
             {
                 "mode": "rt",
-                "command": "transcribe",                
+                "command": "transcribe",
                 "multichannel": "test_channel_1,test_channel_2",
                 "diarization": "speaker",
             },

tests/test_client.py

@@ -95,9 +95,9 @@ def test_handlers_called(mock_server, mocker):
 
     with open(path_to_test_resource("ch.wav"), "rb") as audio_stream:
         ws_client.run_synchronously(
-            transcription_config=transcription_config, 
-            stream=audio_stream, 
-            audio_settings=audio_settings
+            transcription_config=transcription_config,
+            stream=audio_stream,
+            audio_settings=audio_settings,
         )
     mock_server.wait_for_clean_disconnects()
 
@@ -142,7 +142,7 @@ def language_changing_middleware(msg, is_binary):
             transcription_config=transcription_config,
             stream=audio_stream,
             audio_settings=audio_settings,
-            )
+        )
     mock_server.wait_for_clean_disconnects()
 
     # Each handler should have been called once for every message
@@ -181,7 +181,7 @@ def session_ender(event):
             transcription_config=transcription_config,
             stream=audio_stream,
             audio_settings=audio_settings,
-            )
+        )
     mock_server.wait_for_clean_disconnects()
 
     # Only one message should have been sent from the server
@@ -208,7 +208,7 @@ def test_run_synchronously_with_timeout(mock_server):
                 transcription_config=transcription_config,
                 stream=audio_stream,
                 audio_settings=audio_settings,
-                timeout = 0.0001,
+                timeout=0.0001,
             )
 
 
@@ -269,7 +269,7 @@ def session_ender(event, _):
             transcription_config=transcription_config,
             stream=audio_stream,
             audio_settings=audio_settings,
-            )
+        )
     mock_server.wait_for_clean_disconnects()
 
     assert len(mock_server.messages_received) == expect_received_count
@@ -293,7 +293,7 @@ def config_updater(msg):  # pylint: disable=unused-argument
             transcription_config=transcription_config,
             stream=audio_stream,
             audio_settings=audio_settings,
-            )
+        )
     mock_server.wait_for_clean_disconnects()
 
     set_recognition_config_msgs = mock_server.find_messages_by_type(
@@ -320,7 +320,7 @@ def test_start_recognition_sends_speaker_diarization_config(mock_server):
             transcription_config=transcription_config,
             stream=audio_stream,
             audio_settings=audio_settings,
-            )
+        )
     mock_server.wait_for_clean_disconnects()
 
     start_recognition_msgs = mock_server.find_messages_by_type("StartRecognition")
@@ -363,7 +363,7 @@ def stopper(msg):  # pylint: disable=unused-argument
             transcription_config=transcription_config,
             stream=audio_stream,
             audio_settings=audio_settings,
-            )
+        )
     mock_server.wait_for_clean_disconnects()
 
     num_messages_after_stop = len(mock_server.messages_received)
@@ -391,9 +391,9 @@ async def mock_connect(*_, **__):
         with patch.object(client.LOGGER, "error", mock_logger_error_method):
             try:
                 ws_client.run_synchronously(
-                transcription_config=TranscriptionConfig(language="en"),
-                stream=MagicMock(),
-                audio_settings=MagicMock(),
+                    transcription_config=TranscriptionConfig(language="en"),
+                    stream=MagicMock(),
+                    audio_settings=MagicMock(),
                 )
             except ConnectionResetError as exc:
                 assert exc is not None
@@ -511,13 +511,20 @@ async def test__producer_happy_path(mocker):
         if index < exp_iters - 1:
             assert msg == index  # from range in mock_read_in_chunks
             exp_current_seq_no += 1
-            cmp_dicts(original_state, state, exp_diffs={"seq_no": {"single": exp_current_seq_no}})
+            cmp_dicts(
+                original_state,
+                state,
+                exp_diffs={"seq_no": {"single": exp_current_seq_no}},
+            )
         else:
             assert msg == json.dumps(
                 {"message": "EndOfStream", "last_seq_no": {"single": exp_final_seq_no}}
             )
-            cmp_dicts(original_state, state, exp_diffs={"seq_no": {"single": exp_current_seq_no}})
-
+            cmp_dicts(
+                original_state,
+                state,
+                exp_diffs={"seq_no": {"single": exp_current_seq_no}},
+            )
 
     assert exp_iters == len(msgs_states)
 
@@ -647,7 +654,7 @@ def test_language_pack_info_is_stored(mock_server):
             transcription_config=transcription_config,
             stream=audio_stream,
             audio_settings=audio_settings,
-            )
+        )
 
     info = ws_client.get_language_pack_info()
     assert info is not None