Add log likelihood to arviz trace objects

[kyleabeauchamp]

I tried to run the loo function from arviz and got an exception because that part of the model was not annotated in the trace:

data_loo = az.loo(tr, pointwise=True)

TypeError: log likelihood not found in inference data object