from importlib.resources import files
from cosmolayer.cosmosac import Component
from matplotlib import pyplot as plt
path = files("cosmolayer.data") / "C=C(N)O.cosmo"
component = Component(path.read_text())
fig, ax = plt.subplots(figsize=(8, 4))
p = component.probabilities
_ = ax.bar(range(len(p)), p)
_ = ax.set_xlabel("Segment type index")
_ = ax.set_ylabel("Probability")
fig.tight_layout()