Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 8 additions & 9 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -28,13 +28,12 @@ dependencies = ["numpy>=2.0", "pytensor>=2.32.0"]
test = ["pytest", "pytest-cov"]

[tool.flit.module]
name = "distributions"
name = "pytensor_distributions"

[project.urls]
source = "https://github.com/pymc-devs/distributions"
tracker = "https://github.com/pymc-devs/distributions/issues"
documentation = "https://distributions.readthedocs.io"
funding = "https://opencollective.com/pymc"
source = "https://github.com/pymc-devs/pytensor-distributions"
tracker = "https://github.com/pymc-devs/pytensor-distributions/issues"
documentation = "https://pytensor-distributions.readthedocs.io"


[tool.pydocstyle]
Expand All @@ -47,7 +46,7 @@ addopts = [
"--strict-markers",
"--strict-config",
"--color=yes",
"--cov=distributions",
"--cov=pytensor_distributions",
"--cov=tests",
"--cov-report=term-missing",
]
Expand Down Expand Up @@ -78,9 +77,9 @@ ignore = [

[tool.ruff.lint.per-file-ignores]
"docs/source/**/*.ipynb" = ["D", "E", "F", "I", "NPY", "PL", "TID", "UP", "W"]
"distributions/__init__.py" = ["I", "F401", "E402", "F403"]
"distributions/tests/**/*" = ["D", "PLR2004", "TID252"]
"distributions/tests/**/*.ipynb" = ["E", "F"]
"pytensor_distributions/__init__.py" = ["I", "F401", "E402", "F403"]
"pytensor_distributions/tests/**/*" = ["D", "PLR2004", "TID252"]
"pytensor_distributions/tests/**/*.ipynb" = ["E", "F"]

[tool.ruff.lint.pydocstyle]
convention = "numpy"
Expand Down
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
"""PyTensor powered distributions."""

__version__ = "0.1.0"
__version__ = "0.1.1"
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import ppf_bounds_cont
from pytensor_distributions.helper import ppf_bounds_cont


def mean(mu, b, kappa):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import pytensor.tensor as pt
from pytensor.tensor.xlogx import xlogx

from distributions.helper import cdf_bounds, ppf_bounds_disc
from pytensor_distributions.helper import cdf_bounds, ppf_bounds_disc


def mean(p):
Expand Down
2 changes: 1 addition & 1 deletion distributions/beta.py → pytensor_distributions/beta.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from pytensor.tensor.special import betaln
from pytensor.tensor.xlogx import xlogy0

from distributions.helper import ppf_bounds_cont
from pytensor_distributions.helper import ppf_bounds_cont


def mean(alpha, beta):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import pytensor.tensor as pt

from distributions.helper import cdf_bounds, discrete_entropy
from distributions.optimization import find_ppf_discrete
from pytensor_distributions.helper import cdf_bounds, discrete_entropy
from pytensor_distributions.optimization import find_ppf_discrete


def mean(n, alpha, beta):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from pytensor.tensor.special import betaln
from pytensor.tensor.xlogx import xlogy0

from distributions.helper import ppf_bounds_cont
from pytensor_distributions.helper import ppf_bounds_cont


def mean(alpha, beta, lower, upper):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
import pytensor.tensor as pt
from pytensor.tensor.xlogx import xlogy0

from distributions.helper import cdf_bounds, discrete_entropy
from distributions.optimization import find_ppf_discrete
from pytensor_distributions.helper import cdf_bounds, discrete_entropy
from pytensor_distributions.optimization import find_ppf_discrete


def mean(n, p):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import pytensor.tensor as pt
from pytensor.tensor.xlogx import xlogx

from distributions.helper import cdf_bounds, ppf_bounds_disc
from pytensor_distributions.helper import cdf_bounds, ppf_bounds_disc


def _normalize_p(p):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import ppf_bounds_cont
from pytensor_distributions.helper import ppf_bounds_cont


def mean(alpha, beta):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import pytensor.tensor as pt
from pytensor.tensor.xlogx import xlogy0

from distributions.helper import cdf_bounds, ppf_bounds_cont
from pytensor_distributions.helper import cdf_bounds, ppf_bounds_cont


def mean(nu):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import pytensor.tensor as pt
from pytensor.tensor.math import gammaln

from distributions import beta
from pytensor_distributions import beta


def mean(alpha):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
import pytensor.tensor as pt
from pytensor.tensor.math import gammaln

from distributions import betabinomial
from pytensor_distributions import betabinomial


def mean(n, a):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import cdf_bounds, discrete_entropy, ppf_bounds_disc
from pytensor_distributions.helper import cdf_bounds, discrete_entropy, ppf_bounds_disc


def mean(lower, upper):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import (
from pytensor_distributions.helper import (
cdf_bounds,
discrete_entropy,
discrete_kurtosis,
Expand Down
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
import pytensor.tensor as pt

from distributions.helper import continuous_entropy, logdiffexp
from distributions.normal import logcdf as normal_logcdf
from distributions.normal import logpdf as normal_logpdf
from distributions.optimization import find_ppf
from pytensor_distributions.helper import continuous_entropy, logdiffexp
from pytensor_distributions.normal import logcdf as normal_logcdf
from pytensor_distributions.normal import logpdf as normal_logpdf
from pytensor_distributions.optimization import find_ppf


def mean(mu, sigma, nu):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import cdf_bounds, ppf_bounds_cont
from pytensor_distributions.helper import cdf_bounds, ppf_bounds_cont


def mean(lam):
Expand Down
2 changes: 1 addition & 1 deletion distributions/gamma.py → pytensor_distributions/gamma.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import ppf_bounds_cont
from pytensor_distributions.helper import ppf_bounds_cont


def mean(alpha, beta):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import cdf_bounds, ppf_bounds_disc
from pytensor_distributions.helper import cdf_bounds, ppf_bounds_disc


def mean(p):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import cdf_bounds, ppf_bounds_cont
from pytensor_distributions.helper import cdf_bounds, ppf_bounds_cont


def mean(mu, beta):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import ppf_bounds_cont
from pytensor_distributions.helper import ppf_bounds_cont


def mean(beta):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import ppf_bounds_cont
from pytensor_distributions.helper import ppf_bounds_cont


def mean(sigma):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,10 @@
from pytensor.tensor.math import betaincinv
from pytensor.tensor.special import betaln

from distributions.halfnormal import cdf as halfnormal_cdf
from distributions.halfnormal import entropy as halfnormal_entropy
from distributions.halfnormal import logpdf as halfnormal_logpdf
from distributions.helper import cdf_bounds, ppf_bounds_cont
from pytensor_distributions.halfnormal import cdf as halfnormal_cdf
from pytensor_distributions.halfnormal import entropy as halfnormal_entropy
from pytensor_distributions.halfnormal import logpdf as halfnormal_logpdf
from pytensor_distributions.helper import cdf_bounds, ppf_bounds_cont


def mean(nu, sigma):
Expand Down
File renamed without changes.
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import pytensor.tensor as pt

from distributions.helper import cdf_bounds, discrete_entropy
from distributions.optimization import find_ppf_discrete
from pytensor_distributions.helper import cdf_bounds, discrete_entropy
from pytensor_distributions.optimization import find_ppf_discrete


def _support_lower(N, k, n):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import cdf_bounds, ppf_bounds_cont
from pytensor_distributions.helper import cdf_bounds, ppf_bounds_cont


def mean(alpha, beta):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import pytensor.tensor as pt
from pytensor.tensor.xlogx import xlogy0

from distributions.helper import cdf_bounds, ppf_bounds_cont, sf_bounds
from pytensor_distributions.helper import cdf_bounds, ppf_bounds_cont, sf_bounds


def mean(a, b):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import cdf_bounds, ppf_bounds_cont
from pytensor_distributions.helper import cdf_bounds, ppf_bounds_cont


def mean(mu, b):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import ppf_bounds_cont
from pytensor_distributions.helper import ppf_bounds_cont


def mean(mu, s):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
import numpy as np
import pytensor.tensor as pt

from distributions.helper import (
from pytensor_distributions.helper import (
cdf_bounds,
ppf_bounds_cont,
)
from distributions.normal import ppf as normal_ppf
from pytensor_distributions.normal import ppf as normal_ppf


def _logit(x):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import cdf_bounds, ppf_bounds_cont
from pytensor_distributions.helper import cdf_bounds, ppf_bounds_cont


def mean(alpha, beta):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import cdf_bounds, ppf_bounds_cont
from pytensor_distributions.helper import cdf_bounds, ppf_bounds_cont


def mean(mu, sigma):
Expand Down
2 changes: 1 addition & 1 deletion distributions/moyal.py → pytensor_distributions/moyal.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import ppf_bounds_cont
from pytensor_distributions.helper import ppf_bounds_cont


def mean(mu, sigma):
Expand Down
File renamed without changes.
File renamed without changes.
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
import pytensor.tensor as pt
from pytensor.tensor.math import gammaln

from distributions.mvnormal import quaddist_chol
from pytensor_distributions.mvnormal import quaddist_chol


def mean(nu, mu, cov):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
import pytensor.tensor as pt
from pytensor.tensor.xlogx import xlogy0

from distributions.helper import cdf_bounds, discrete_entropy, sf_bounds
from distributions.optimization import find_ppf_discrete
from pytensor_distributions.helper import cdf_bounds, discrete_entropy, sf_bounds
from pytensor_distributions.optimization import find_ppf_discrete


def mean(n, p):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import ppf_bounds_cont
from pytensor_distributions.helper import ppf_bounds_cont


def mean(mu, sigma):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

import pytensor.tensor as pt

from distributions.helper import ppf_bounds_cont, ppf_bounds_disc
from pytensor_distributions.helper import ppf_bounds_cont, ppf_bounds_disc


def find_ppf(q, lower, upper, cdf, *params):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import ppf_bounds_cont
from pytensor_distributions.helper import ppf_bounds_cont


def mean(alpha, m):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
import pytensor.tensor as pt
from pytensor.tensor.xlogx import xlogy0

from distributions.helper import cdf_bounds, discrete_entropy, sf_bounds
from distributions.optimization import find_ppf_discrete
from pytensor_distributions.helper import cdf_bounds, discrete_entropy, sf_bounds
from pytensor_distributions.optimization import find_ppf_discrete


def mean(mu):
Expand Down
4 changes: 2 additions & 2 deletions distributions/rice.py → pytensor_distributions/rice.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,14 @@
import pytensor.tensor as pt

from distributions.helper import (
from pytensor_distributions.helper import (
cdf_bounds,
continuous_entropy,
continuous_kurtosis,
continuous_mode,
continuous_skewness,
marcum_q1_complement,
)
from distributions.optimization import find_ppf
from pytensor_distributions.optimization import find_ppf


def _laguerre_half(q):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytensor.tensor as pt

from distributions.helper import cdf_bounds, ppf_bounds_cont
from pytensor_distributions.helper import cdf_bounds, ppf_bounds_cont


def mean(nu, tau2):
Expand Down
Loading
Loading