From cf07c153e9af4ab2df0bdc7c07c319b50bf823eb Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 4 Oct 2019 16:55:51 -0400 Subject: [PATCH] tests fix --- tests/test_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_models.py b/tests/test_models.py index cdb59f74fd..d5bf3b6e9b 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -639,7 +639,7 @@ def test_no_val_end_module(): tags_path = logger.experiment.get_data_path(logger.experiment.name, logger.experiment.version) tags_path = os.path.join(tags_path, 'meta_tags.csv') model_2 = LightningTestModel.load_from_metrics(weights_path=new_weights_path, - tags_csv=tags_path, on_gpu=False) + tags_csv=tags_path) model_2.eval() # make prediction