parameter name fix

pyro-ppl · Sep 19, 2024 · a19d51d · a19d51d
1 parent 57f81b7
commit a19d51d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyro/distributions/transforms/householder.py b/pyro/distributions/transforms/householder.py
@@ -29,7 +29,7 @@ def __init__(self, u_unnormed=None):
     # Construct normalized vectors for Householder transform
     def u(self):
         u_unnormed = self.u_unnormed() if callable(self.u_unnormed) else self.u_unnormed
-        norm = torch.linalg.norm(u_unnormed, p=2, dim=-1, keepdim=True)
+        norm = torch.linalg.norm(u_unnormed, ord=2, dim=-1, keepdim=True)
         return torch.div(u_unnormed, norm)
 
     def _call(self, x):