How to plot multiple scatterplots from pymc3 bayesian hierarchical model trace with 2 predictors and by groups

Question

I have a dataset structured like attached toy_data.csv. I would like to grab the relevant batch & predictor from the hierarchical model trace to plot them and fit the abline