Make test more explicit

pyro-ppl · Apr 22, 2021 · 5dfee9e · 5dfee9e
1 parent 3676b31
commit 5dfee9e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/optim/test_optim.py b/tests/optim/test_optim.py
@@ -284,6 +284,5 @@ def optim_config(param_name):
     param_names.clear()
     pyro.clear_param_store()
 
-    print(expected_param_names)
-    print(actual_param_names)
+    assert actual_param_names == {"scale", "loc.weight", "loc.bias"}
     assert actual_param_names == expected_param_names