From 2e208f608cb83681d6156de50903b9679b19c1da Mon Sep 17 00:00:00 2001 From: Somshubra Majumdar Date: Wed, 13 May 2020 13:35:33 -0700 Subject: [PATCH] Style fixes in tests Signed-off-by: smajumdar --- tests/integration/test_asr_gradient_step_and_eval.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/integration/test_asr_gradient_step_and_eval.py b/tests/integration/test_asr_gradient_step_and_eval.py index 08ba594d8b66..d68898c076b6 100644 --- a/tests/integration/test_asr_gradient_step_and_eval.py +++ b/tests/integration/test_asr_gradient_step_and_eval.py @@ -242,7 +242,7 @@ def test_contextnet_ctc_training(self): feat_in=contextnet_model_definition['AudioToMelSpectrogramPreprocessor']['features'], **contextnet_model_definition['ContextNetEncoder'], ) - contextnet_decoder = nemo_asr.ContextNetDecoderForCTC(feat_in=32, hidden_size=16, num_classes=len(self.labels)) + contextnet_decoder = nemo_asr.ContextNetDecoderForCTC(feat_in=32, hidden_size=16, num_classes=len(self.labels)) ctc_loss = nemo_asr.CTCLossNM(num_classes=len(self.labels)) # DAG