In the "replace_latents" function, the second 'elif' condition should be : elif name == "**w2**": rv_kwargs["value"] = w2 Instead of : elif name == "_w_": rv_kwargs["value"] = w2
In the "replace_latents" function, the second 'elif' condition should be :
elif name == "w2":
rv_kwargs["value"] = w2
Instead of :
elif name == "w":
rv_kwargs["value"] = w2