Skip to content

Commit 05ce461

Browse files
committed
reapply format
1 parent c045524 commit 05ce461

File tree

4 files changed

+10
-5
lines changed

4 files changed

+10
-5
lines changed

tests/models/autoencoders/test_models_vae.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -376,10 +376,12 @@ def prepare_init_args_and_inputs_for_common(self):
376376
return self.init_dict, self.inputs_dict()
377377

378378
@unittest.skip
379-
def test_training(self): ...
379+
def test_training(self):
380+
...
380381

381382
@unittest.skip
382-
def test_ema_training(self): ...
383+
def test_ema_training(self):
384+
...
383385

384386

385387
class AutoencoderKLTemporalDecoderFastTests(ModelTesterMixin, unittest.TestCase):

tests/pipelines/amused/test_amused.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,8 @@ def test_inference_batch_consistent(self, batch_sizes=[2]):
124124
self._test_inference_batch_consistent(batch_sizes=batch_sizes, batch_generator=False)
125125

126126
@unittest.skip("aMUSEd does not support lists of generators")
127-
def test_inference_batch_single_identical(self): ...
127+
def test_inference_batch_single_identical(self):
128+
...
128129

129130

130131
@slow

tests/pipelines/amused/test_amused_img2img.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,8 @@ def test_inference_batch_consistent(self, batch_sizes=[2]):
126126
self._test_inference_batch_consistent(batch_sizes=batch_sizes, batch_generator=False)
127127

128128
@unittest.skip("aMUSEd does not support lists of generators")
129-
def test_inference_batch_single_identical(self): ...
129+
def test_inference_batch_single_identical(self):
130+
...
130131

131132

132133
@slow

tests/pipelines/amused/test_amused_inpaint.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,8 @@ def test_inference_batch_consistent(self, batch_sizes=[2]):
130130
self._test_inference_batch_consistent(batch_sizes=batch_sizes, batch_generator=False)
131131

132132
@unittest.skip("aMUSEd does not support lists of generators")
133-
def test_inference_batch_single_identical(self): ...
133+
def test_inference_batch_single_identical(self):
134+
...
134135

135136

136137
@slow

0 commit comments

Comments
 (0)