from stratmc.data import load_object, load_trace
from stratmc.plotting import sample_ages_per_chain

full_trace = load_trace('examples/example_docs_trace')
example_sample_path = 'examples/example_sample_df'
sample_df = load_object(example_sample_path)
section = '1'

sample_ages_per_chain(full_trace, sample_df, section, chains = [0, 1, 2, 3])

plt.show()