Problem in Implementing a Graphical Model Using Pyro

Question

I am trying to implement this graphical model using Pyro:

My implementation is:

def model(data): 
    p = pyro.sample(\'p\', dist.Beta(1, 1))