Fix tests

2021-07-26 16:35:45 +02:00
parent 69359b5af3
commit 5e89a83c88
4 changed files with 10 additions and 10 deletions
--- a/tests/nets/test_deepsets.py
+++ b/tests/nets/test_deepsets.py
@@ -41,7 +41,7 @@ def test_phi():
    y = phi(torch.rand(input_dim))
    y = phi(torch.rand(7,7,7,input_dim))

-    assert len(phi.parameters() > 0)
+    assert len(list(phi.parameters())) > 0

 def test_deepsets():
    m = ds.DeepSetsModule.from_config(ds_config)