Linting and formatting.

pyro-ppl · Jun 27, 2024 · f995b97 · f995b97
1 parent 49ca388
commit f995b97
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/infer/test_util.py b/tests/infer/test_util.py
@@ -76,9 +76,9 @@ def guide(zdim=1, scale=1.0):
 
 def test_render_model_deterministic_param():
     def model():
-        value = pyro.param("param", torch.tensor(0.))
+        value = pyro.param("param", torch.tensor(0.0))
         pyro.deterministic("deterministic", value)
 
     graph = pyro.render_model(model, render_params=True, render_deterministic=True)
 
-    assert '\tparam -> deterministic\n' in graph.body
+    assert "\tparam -> deterministic\n" in graph.body