@@ -87,7 +87,6 @@ def test_shape(self, input_param, input_shape, expected_shape, expected_latent_s
8787 self .assertEqual (result [2 ].shape , expected_latent_shape )
8888
8989 @parameterized .expand (CASES )
90-
9190 def test_shape_with_convtranspose_and_checkpointing (
9291 self , input_param , input_shape , expected_shape , expected_latent_shape
9392 ):
@@ -152,7 +151,6 @@ def test_shape_reconstruction(self):
152151 result = net .reconstruct (torch .randn (input_shape ).to (device ))
153152 self .assertEqual (result .shape , expected_shape )
154153
155-
156154 def test_shape_reconstruction_with_convtranspose_and_checkpointing (self ):
157155 input_param , input_shape , expected_shape , _ = CASES [0 ]
158156 input_param = input_param .copy ()
@@ -170,7 +168,6 @@ def test_shape_encode(self):
170168 self .assertEqual (result [0 ].shape , expected_latent_shape )
171169 self .assertEqual (result [1 ].shape , expected_latent_shape )
172170
173-
174171 def test_shape_encode_with_convtranspose_and_checkpointing (self ):
175172 input_param , input_shape , _ , expected_latent_shape = CASES [0 ]
176173 input_param = input_param .copy ()
@@ -190,7 +187,6 @@ def test_shape_sampling(self):
190187 )
191188 self .assertEqual (result .shape , expected_latent_shape )
192189
193-
194190 def test_shape_sampling_convtranspose_and_checkpointing (self ):
195191 input_param , _ , _ , expected_latent_shape = CASES [0 ]
196192 input_param = input_param .copy ()
@@ -209,7 +205,6 @@ def test_shape_decode(self):
209205 result = net .decode (torch .randn (latent_shape ).to (device ))
210206 self .assertEqual (result .shape , expected_input_shape )
211207
212-
213208 def test_shape_decode_convtranspose_and_checkpointing (self ):
214209 input_param , expected_input_shape , _ , latent_shape = CASES [0 ]
215210 input_param = input_param .copy ()
0 commit comments