PyAETNA 4.0.0__py3-none-any.whl

PyAETNA/cli.py

@@ -0,0 +1,52 @@
+"""Command-line entry point: ``pyaetna config.yaml [more.yaml ...]``.
+
+Also reachable as ``python -m PyAETNA config.yaml``.
+"""
+
+from __future__ import annotations
+
+import argparse
+from collections.abc import Sequence
+
+
+def main(argv: Sequence[str] | None = None) -> None:
+    """Run the PyAETNA pipeline on one or more YAML configuration files."""
+    parser = argparse.ArgumentParser(
+        prog="pyaetna",
+        description="Bayesian modelling of exoplanet light curves and radial velocities.",
+    )
+    parser.add_argument(
+        "configs", nargs="+", metavar="CONFIG", help="YAML configuration file(s)"
+    )
+    parser.add_argument("--version", action="version", version=_version())
+    args = parser.parse_args(argv)
+
+    # Both must happen before JAX / pyplot are first imported, hence the
+    # imports inside the function; _device_config also limits BLAS threads,
+    # so it must come before anything that imports NumPy (matplotlib does).
+    import matplotlib
+
+    from . import _device_config  # noqa: F401  (XLA_FLAGS, BLAS threads for CPU chains)
+
+    matplotlib.use("PDF")
+
+    from .run import run
+    from .sampling.engine.exceptions import MultiplePeaksFound
+
+    stopped = False
+    for config in args.configs:
+        print("processing:", config)
+        try:
+            run(config)
+        except MultiplePeaksFound as exc:
+            # not a failure: the run stopped as asked (sampling.on_multiple_peaks: stop)
+            print(f"stopped: {exc}")
+            stopped = True
+    if stopped:
+        raise SystemExit(3)
+
+
+def _version() -> str:
+    from .version import __version__
+
+    return f"PyAETNA {__version__}"