Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Un-skip some Horovod tests (#8676)

This commit is contained in:
Carlos Mocholí 2021-08-02 17:54:05 +02:00 committed by GitHub
parent 850416f0a0
commit d187008e84
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/test_horovod.py
View File

@ -242,8 +242,6 @@ def test_horovod_transfer_batch_to_gpu(tmpdir):
tpipes.run_model_test_without_loggers(trainer_options, model) tpipes.run_model_test_without_loggers(trainer_options, model)
# todo: need to be fixed :]
@pytest.mark.skip("TODO: flaky test - Fatal Python error: Aborted")
@RunIf(skip_windows=True, horovod=True) @RunIf(skip_windows=True, horovod=True)
def test_horovod_multi_optimizer(tmpdir): def test_horovod_multi_optimizer(tmpdir):
model = BasicGAN() model = BasicGAN()
@ -377,8 +375,6 @@ def test_accuracy_metric_horovod():
horovod.run(_compute_batch, np=2) horovod.run(_compute_batch, np=2)
# todo: need to be fixed :]
@pytest.mark.skip("TODO: flaky test - Fatal Python error: Aborted")
@RunIf(skip_windows=True, horovod=True) @RunIf(skip_windows=True, horovod=True)
def test_horovod_multi_optimizer_with_scheduling_stepping(tmpdir): def test_horovod_multi_optimizer_with_scheduling_stepping(tmpdir):
class TestModel(BoringModel): class TestModel(BoringModel):