Fix failing GPU tests (#722)

* Fix distributed_backend=None test

We now throw a warning instead of an exception. Update test
to reflect this.

* Fix test_tube logger close when debug=True
This commit is contained in:
Nic Eggert
2020-01-21 14:26:43 -05:00
committed by William Falcon
parent ca894f081b
commit dfb6d3626e
2 changed files with 4 additions and 3 deletions
+3 -2
View File
@@ -135,8 +135,9 @@ class TestTubeLogger(LightningLoggerBase):
def close(self):
# TODO: HACK figure out where this is being set to true
self.experiment.debug = self.debug
exp = self.experiment
exp.close()
if not self.debug:
exp = self.experiment
exp.close()
@property
def rank(self):
+1 -1
View File
@@ -183,7 +183,7 @@ def test_multi_gpu_none_backend(tmpdir):
gpus='-1'
)
with pytest.raises(MisconfigurationException):
with pytest.warns(UserWarning):
tutils.run_model_test(trainer_options, model)