From 9ea287f8ec22867343bcef6e54d7732042484944 Mon Sep 17 00:00:00 2001 From: jerryli27 Date: Fri, 11 Oct 2024 20:37:10 +0000 Subject: [PATCH] Style changes to pass the precommits --- moshi/moshi/modules/conv_test.py | 6 +++--- moshi/moshi/modules/seanet_test.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/moshi/moshi/modules/conv_test.py b/moshi/moshi/modules/conv_test.py index 1d95237..01620b5 100644 --- a/moshi/moshi/modules/conv_test.py +++ b/moshi/moshi/modules/conv_test.py @@ -10,7 +10,7 @@ CONV1D_DATA = [ - # batch_size,in_channels,out_channels,seq_len,kernel_size + # batch_size, in_channels, out_channels, seq_len, kernel_size pytest.param( 3, 4, 5, 10, 6, id='small conv1d test 1', @@ -30,7 +30,7 @@ ] CONV1D_TRANSPOSE_DATA = [ - # batch_size,in_channels,out_channels,seq_len,kernel_size,stride + # batch_size, in_channels, out_channels, seq_len, kernel_size, stride pytest.param( 3, 4, 5, 10, 6, 1, id='small conv1d transpose test 1', @@ -60,7 +60,7 @@ def _init_weights(module, generator=None): nn.init.xavier_uniform_(param, generator=generator) -@pytest.mark.parametrize("batch_size,in_channels,out_channels,seq_len,kernel_size", CONV1D_DATA) +@pytest.mark.parametrize("batch_size, in_channels, out_channels, seq_len, kernel_size", CONV1D_DATA) def test_conv1d(batch_size, in_channels, out_channels, seq_len, kernel_size): """Test that StreamingConv1d() calls are causal. Having new inputs does not change the previous output.""" assert seq_len > kernel_size diff --git a/moshi/moshi/modules/seanet_test.py b/moshi/moshi/modules/seanet_test.py index 35fe1b5..0a8aa61 100644 --- a/moshi/moshi/modules/seanet_test.py +++ b/moshi/moshi/modules/seanet_test.py @@ -10,7 +10,7 @@ SEANET_RESNET_DATA = [ - # batch_size,dim,res_layer_index,seq_len,kernel_size + # batch_size, dim, res_layer_index, seq_len, kernel_size pytest.param( 3, 4, 1, 10, 6, id='small resnet test 1',