Fix typo in reparam.py

pyro-ppl · Jul 2, 2024 · ca732fe · ca732fe
1 parent 92c6274
commit ca732fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/numpyro/infer/reparam.py b/numpyro/infer/reparam.py
@@ -226,7 +226,7 @@ class NeuTraReparam(Reparam):
 
         # Step 2. Use trained guide in NeuTra MCMC
         neutra = NeuTraReparam(guide)
-        model = netra.reparam(model)
+        model = neutra.reparam(model)
         nuts = NUTS(model)
         # ...now use the model in HMC or NUTS...