Implement suggestions, remove whitespace

Author: ddrevicky

tests/trainer/test_trainer.py

@@ -1181,16 +1181,16 @@ def test_row_log_interval(log_metrics_mock, tmpdir, train_batches, max_steps, lo
     expected_calls = [call(metrics=ANY, step=s) for s in range(log_interval - 1, max_steps, log_interval)]
     log_metrics_mock.assert_has_calls(expected_calls)
 
-    
-@pytest.mark.parametrize(['input_arg', 'expected'], [
+
+@pytest.mark.parametrize(['profiler', 'expected'], [
     (None, PassThroughProfiler),
     (SimpleProfiler(), SimpleProfiler),
     (AdvancedProfiler(), AdvancedProfiler),
     ('simple', SimpleProfiler),
     ('advanced', AdvancedProfiler),
 ])
 def test_trainer_profiler_correct_args(input_arg, expected):
-    kwargs = {'profiler': input_arg} if input_arg is not None else {}
+    kwargs = {'profiler': profiler} if profiler is not None else {}
     trainer = Trainer(**kwargs)
     assert isinstance(trainer.profiler, expected)