clear skipping tests (#1285)

* clear skipping tests

* fix simple/multi GPU

* review: simplify
This commit is contained in:
Jirka Borovec
2020-03-30 18:29:23 -04:00
committed by GitHub
parent 31a658e558
commit 2ca5356429
5 changed files with 15 additions and 49 deletions
+1 -3
View File
@@ -206,12 +206,10 @@ def test_disabled_validation():
assert model.validation_end_invoked, 'did not run `validation_end` with `fast_dev_run=True`'
@pytest.mark.skipif(not torch.cuda.is_available(), reason="test requires GPU machine")
def test_single_gpu_batch_parse():
tutils.reset_seed()
if not tutils.can_run_gpu_test():
return
trainer = Trainer()
# batch is just a tensor