Backport PR gammapy#5576: Add UniformPrior and GaussianPrior in `…

…gammapy.modeling.models __all__`
meeseeksmachine · Nov 18, 2024 · 1b54437 · 1b54437
1 parent f3fdd35
commit 1b54437
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/gammapy/modeling/models/__init__.py b/gammapy/modeling/models/__init__.py
@@ -1,5 +1,6 @@
 # Licensed under a 3-clause BSD style license - see LICENSE.rst
 """Built-in models in Gammapy."""
+
 from gammapy.utils.registry import Registry
 from .core import DatasetModels, Model, ModelBase, Models
 from .cube import (
@@ -109,6 +110,8 @@
     "PowerLawSpectralModel",
     "PowerLawTemporalModel",
     "Prior",
+    "GaussianPrior",
+    "UniformPrior",
     "scale_plot_flux",
     "ScaleSpectralModel",
     "Shell2SpatialModel",

diff --git a/gammapy/modeling/models/prior.py b/gammapy/modeling/models/prior.py
@@ -1,12 +1,13 @@
 # Licensed under a 3-clause BSD style license - see LICENSE.rst
 """Priors for Gammapy."""
+
 import logging
 import numpy as np
 import astropy.units as u
 from gammapy.modeling import PriorParameter, PriorParameters
 from .core import ModelBase
 
-__all__ = ["GaussianPrior", "UniformPrior"]
+__all__ = ["GaussianPrior", "UniformPrior", "Prior"]
 
 log = logging.getLogger(__name__)