import importlib.util

import ergodic

# 1. Print the installed version and the graph format it writes.
print("ergodic", ergodic.__version__, "writes", ergodic.GRAPH_FORMAT)

# 2. Check which optional packages this environment has: found or missing.
optional = {
    "pymc": "bayes, scm, forecast and latent",
    "jax": "faster sampling, with numpyro",
    "numpyro": "faster sampling, with jax",
    "welt": "welt",
}
for module, use in optional.items():
    status = "found" if importlib.util.find_spec(module) else "missing"
    print(f"{module:8} {status:8} {use}")