Fix deprecation warning in sigmoid

2021-07-20 12:52:02 +02:00
parent a9c6857b5b
commit ee7b6b607f
2 changed files with 8 additions and 6 deletions
--- a/tests/nets/test_deepsets.py
+++ b/tests/nets/test_deepsets.py
@@ -24,7 +24,11 @@ def test_constructor():
    phi_config["output_dim"] = 2
    phi_config["final_activation"] = "sigmoid"
    phi = ds.Phi.from_config(phi_config)
-    assert phi.final_activation == torch.nn.functional.sigmoid
+    assert phi.final_activation == torch.sigmoid
+
+    phi_config["final_activation"] = "relu"
+    phi = ds.Phi.from_config(phi_config)
+    assert phi.final_activation == torch.nn.functional.relu

 def test_phi():
    input_dim = 5