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

full_trace = load_trace('examples/example_docs_trace')
example_sample_path = 'examples/example_sample_df'
example_ages_path = 'examples/example_ages_df'
sample_df = load_object(example_sample_path)
ages_df = load_object(example_ages_path)

age_height_model(sample_df, ages_df, full_trace)

plt.show()