fixed hpc save, load. cleaned apu

2019-07-26 22:13:06 -04:00 · 2019-07-26 22:13:06 -04:00 · a6ae97ac09
parent 348223a702
commit a6ae97ac09
1 changed files with 1 additions and 1 deletions
--- a/tests/test_models.py
+++ b/tests/test_models.py
@ -115,7 +115,7 @@ def test_model_saving_loading():
    # make prediction
    # assert that both predictions are the same
    new_pred = model_2(x)
-    assert torch.eq(pred_before_saving, new_pred)
+    assert torch.all(torch.eq(pred_before_saving, new_pred)).item() == 1

    clear_save_dir()