import matplotlib.pyplot as plt
plt.clf()
from tf_pwa.utils import plot_particle_model
axis = plot_particle_model("BW")