Browse Source

remove debugprint

justheuristic 5 years ago
parent
commit
cb1909a3bb
1 changed files with 0 additions and 1 deletions
  1. 0 1
      tests/test_moe.py

+ 0 - 1
tests/test_moe.py

@@ -63,7 +63,6 @@ def test_compute_expert_scores():
 
         for b in range(len(ii)):
             for e in range(len(ii[b])):
-                print(end='.')
                 assert torch.allclose(logits[b, e], gx[b, ii[b][e]] + gy[b, jj[b][e]]), "compute_expert_scores returned incorrect score"
 
         dht.shutdown()