From 486a841dabfc38a3b5e70194d8c13e61cd2dcb41 Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Wed, 11 Dec 2019 06:50:56 -0800
Subject: [PATCH] remove summary

---
 tests/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/utils.py b/tests/utils.py
index 8bdf55af..7465b396 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -159,7 +159,7 @@ def run_prediction(dataloader, trained_model, dp=False, min_acc=0.50):
         acc = torch.mean(acc).item()
 
     else:
-        y_hat = trained_model(x, 0)
+        y_hat = trained_model(x)
 
         # acc
         labels_hat = torch.argmax(y_hat, dim=1)