tests fix

2019-10-04 16:55:51 -04:00 · 2019-10-04 16:55:51 -04:00 · cf07c153e9
parent 36f0b5bbd0
commit cf07c153e9
1 changed files with 1 additions and 1 deletions
--- 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