Browse Source

[to #42322933] disable unstable trainer test

master
Yingda Chen 3 years ago
parent
commit
089cadab4b
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      tests/trainers/test_image_denoise_trainer.py

+ 2
- 2
tests/trainers/test_image_denoise_trainer.py View File

@@ -51,7 +51,7 @@ class ImageDenoiseTrainerTest(unittest.TestCase):
shutil.rmtree(self.tmp_dir, ignore_errors=True)
super().tearDown()

@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
def test_trainer(self):
kwargs = dict(
model=self.model_id,
@@ -65,7 +65,7 @@ class ImageDenoiseTrainerTest(unittest.TestCase):
for i in range(2):
self.assertIn(f'epoch_{i+1}.pth', results_files)

@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
def test_trainer_with_model_and_args(self):
model = NAFNetForImageDenoise.from_pretrained(self.cache_path)
kwargs = dict(


Loading…
Cancel
Save