fix lint

harvardnlp · Apr 9, 2020 · 08b53c5 · 08b53c5
1 parent a792af7
commit 08b53c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_struct/test_distributions.py b/torch_struct/test_distributions.py
@@ -46,8 +46,8 @@ def test_simple(data, seed):
     marginals = dist.marginals
     assert ((samples.mean(0) - marginals).abs() < 0.2).all()
 
-    kmax = dist.kmax(5)
-    count = dist.count
+    dist.kmax(5)
+    dist.count
 
 
 @given(data(), integers(min_value=1, max_value=20))