diff --git a/pyproject.toml b/pyproject.toml index 1fe02a4..835ae57 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -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] @@ -47,7 +46,7 @@ addopts = [ "--strict-markers", "--strict-config", "--color=yes", - "--cov=distributions", + "--cov=pytensor_distributions", "--cov=tests", "--cov-report=term-missing", ] @@ -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" diff --git a/distributions/__init__.py b/pytensor_distributions/__init__.py similarity index 63% rename from distributions/__init__.py rename to pytensor_distributions/__init__.py index 64cee6a..fed99cf 100644 --- a/distributions/__init__.py +++ b/pytensor_distributions/__init__.py @@ -1,3 +1,3 @@ """PyTensor powered distributions.""" -__version__ = "0.1.0" +__version__ = "0.1.1" diff --git a/distributions/asymmetriclaplace.py b/pytensor_distributions/asymmetriclaplace.py similarity index 97% rename from distributions/asymmetriclaplace.py rename to pytensor_distributions/asymmetriclaplace.py index 4c10f8e..4dc1a0e 100644 --- a/distributions/asymmetriclaplace.py +++ b/pytensor_distributions/asymmetriclaplace.py @@ -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): diff --git a/distributions/bernoulli.py b/pytensor_distributions/bernoulli.py similarity index 94% rename from distributions/bernoulli.py rename to pytensor_distributions/bernoulli.py index d64beb9..cbf39b9 100644 --- a/distributions/bernoulli.py +++ b/pytensor_distributions/bernoulli.py @@ -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): diff --git a/distributions/beta.py b/pytensor_distributions/beta.py similarity index 98% rename from distributions/beta.py rename to pytensor_distributions/beta.py index aa0afaf..fd04b74 100644 --- a/distributions/beta.py +++ b/pytensor_distributions/beta.py @@ -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): diff --git a/distributions/betabinomial.py b/pytensor_distributions/betabinomial.py similarity index 96% rename from distributions/betabinomial.py rename to pytensor_distributions/betabinomial.py index 747fd87..71b5031 100644 --- a/distributions/betabinomial.py +++ b/pytensor_distributions/betabinomial.py @@ -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): diff --git a/distributions/betascaled.py b/pytensor_distributions/betascaled.py similarity index 98% rename from distributions/betascaled.py rename to pytensor_distributions/betascaled.py index 7864c99..47709e1 100644 --- a/distributions/betascaled.py +++ b/pytensor_distributions/betascaled.py @@ -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): diff --git a/distributions/binomial.py b/pytensor_distributions/binomial.py similarity index 90% rename from distributions/binomial.py rename to pytensor_distributions/binomial.py index 1717d4a..b0a72ee 100644 --- a/distributions/binomial.py +++ b/pytensor_distributions/binomial.py @@ -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): diff --git a/distributions/categorical.py b/pytensor_distributions/categorical.py similarity index 97% rename from distributions/categorical.py rename to pytensor_distributions/categorical.py index 4c15f9a..0cf0a7d 100644 --- a/distributions/categorical.py +++ b/pytensor_distributions/categorical.py @@ -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): diff --git a/distributions/cauchy.py b/pytensor_distributions/cauchy.py similarity index 96% rename from distributions/cauchy.py rename to pytensor_distributions/cauchy.py index b1c8eec..b78be52 100644 --- a/distributions/cauchy.py +++ b/pytensor_distributions/cauchy.py @@ -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): diff --git a/distributions/chisquared.py b/pytensor_distributions/chisquared.py similarity index 95% rename from distributions/chisquared.py rename to pytensor_distributions/chisquared.py index e73ead1..efb77f9 100644 --- a/distributions/chisquared.py +++ b/pytensor_distributions/chisquared.py @@ -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): diff --git a/distributions/dirichlet.py b/pytensor_distributions/dirichlet.py similarity index 97% rename from distributions/dirichlet.py rename to pytensor_distributions/dirichlet.py index 87b9f01..a2637fe 100644 --- a/distributions/dirichlet.py +++ b/pytensor_distributions/dirichlet.py @@ -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): diff --git a/distributions/dirichletmultinomial.py b/pytensor_distributions/dirichletmultinomial.py similarity index 97% rename from distributions/dirichletmultinomial.py rename to pytensor_distributions/dirichletmultinomial.py index 48368f0..2cff9d1 100644 --- a/distributions/dirichletmultinomial.py +++ b/pytensor_distributions/dirichletmultinomial.py @@ -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): diff --git a/distributions/discreteuniform.py b/pytensor_distributions/discreteuniform.py similarity index 94% rename from distributions/discreteuniform.py rename to pytensor_distributions/discreteuniform.py index 4f51a2c..d1c8d23 100644 --- a/distributions/discreteuniform.py +++ b/pytensor_distributions/discreteuniform.py @@ -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): diff --git a/distributions/discreteweibull.py b/pytensor_distributions/discreteweibull.py similarity index 97% rename from distributions/discreteweibull.py rename to pytensor_distributions/discreteweibull.py index 1329415..99eb919 100644 --- a/distributions/discreteweibull.py +++ b/pytensor_distributions/discreteweibull.py @@ -1,6 +1,6 @@ import pytensor.tensor as pt -from distributions.helper import ( +from pytensor_distributions.helper import ( cdf_bounds, discrete_entropy, discrete_kurtosis, diff --git a/distributions/exgaussian.py b/pytensor_distributions/exgaussian.py similarity index 90% rename from distributions/exgaussian.py rename to pytensor_distributions/exgaussian.py index 3de3447..d314c65 100644 --- a/distributions/exgaussian.py +++ b/pytensor_distributions/exgaussian.py @@ -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): diff --git a/distributions/exponential.py b/pytensor_distributions/exponential.py similarity index 95% rename from distributions/exponential.py rename to pytensor_distributions/exponential.py index 1cdf886..3905030 100644 --- a/distributions/exponential.py +++ b/pytensor_distributions/exponential.py @@ -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): diff --git a/distributions/gamma.py b/pytensor_distributions/gamma.py similarity index 97% rename from distributions/gamma.py rename to pytensor_distributions/gamma.py index cdaa8f8..160561b 100644 --- a/distributions/gamma.py +++ b/pytensor_distributions/gamma.py @@ -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): diff --git a/distributions/geometric.py b/pytensor_distributions/geometric.py similarity index 94% rename from distributions/geometric.py rename to pytensor_distributions/geometric.py index 1116917..7ef8a1c 100644 --- a/distributions/geometric.py +++ b/pytensor_distributions/geometric.py @@ -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): diff --git a/distributions/gumbel.py b/pytensor_distributions/gumbel.py similarity index 95% rename from distributions/gumbel.py rename to pytensor_distributions/gumbel.py index b324d09..ead5a34 100644 --- a/distributions/gumbel.py +++ b/pytensor_distributions/gumbel.py @@ -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): diff --git a/distributions/halfcauchy.py b/pytensor_distributions/halfcauchy.py similarity index 95% rename from distributions/halfcauchy.py rename to pytensor_distributions/halfcauchy.py index ff8fac1..7c1b4fd 100644 --- a/distributions/halfcauchy.py +++ b/pytensor_distributions/halfcauchy.py @@ -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): diff --git a/distributions/halfnormal.py b/pytensor_distributions/halfnormal.py similarity index 95% rename from distributions/halfnormal.py rename to pytensor_distributions/halfnormal.py index 697ffe7..efc76ad 100644 --- a/distributions/halfnormal.py +++ b/pytensor_distributions/halfnormal.py @@ -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): diff --git a/distributions/halfstudentt.py b/pytensor_distributions/halfstudentt.py similarity index 92% rename from distributions/halfstudentt.py rename to pytensor_distributions/halfstudentt.py index 074981b..010bbd1 100644 --- a/distributions/halfstudentt.py +++ b/pytensor_distributions/halfstudentt.py @@ -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): diff --git a/distributions/helper.py b/pytensor_distributions/helper.py similarity index 100% rename from distributions/helper.py rename to pytensor_distributions/helper.py diff --git a/distributions/hypergeometric.py b/pytensor_distributions/hypergeometric.py similarity index 95% rename from distributions/hypergeometric.py rename to pytensor_distributions/hypergeometric.py index a60e5aa..3f9d1c0 100644 --- a/distributions/hypergeometric.py +++ b/pytensor_distributions/hypergeometric.py @@ -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): diff --git a/distributions/inversegamma.py b/pytensor_distributions/inversegamma.py similarity index 97% rename from distributions/inversegamma.py rename to pytensor_distributions/inversegamma.py index ecc0e51..30fe819 100644 --- a/distributions/inversegamma.py +++ b/pytensor_distributions/inversegamma.py @@ -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): diff --git a/distributions/kumaraswamy.py b/pytensor_distributions/kumaraswamy.py similarity index 96% rename from distributions/kumaraswamy.py rename to pytensor_distributions/kumaraswamy.py index dbb72a3..7e8111d 100644 --- a/distributions/kumaraswamy.py +++ b/pytensor_distributions/kumaraswamy.py @@ -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): diff --git a/distributions/laplace.py b/pytensor_distributions/laplace.py similarity index 96% rename from distributions/laplace.py rename to pytensor_distributions/laplace.py index 7bf9209..3a250c9 100644 --- a/distributions/laplace.py +++ b/pytensor_distributions/laplace.py @@ -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): diff --git a/distributions/logistic.py b/pytensor_distributions/logistic.py similarity index 96% rename from distributions/logistic.py rename to pytensor_distributions/logistic.py index f071d6f..e0766fc 100644 --- a/distributions/logistic.py +++ b/pytensor_distributions/logistic.py @@ -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): diff --git a/distributions/logitnormal.py b/pytensor_distributions/logitnormal.py similarity index 97% rename from distributions/logitnormal.py rename to pytensor_distributions/logitnormal.py index a2ff358..9f9903c 100644 --- a/distributions/logitnormal.py +++ b/pytensor_distributions/logitnormal.py @@ -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): diff --git a/distributions/loglogistic.py b/pytensor_distributions/loglogistic.py similarity index 97% rename from distributions/loglogistic.py rename to pytensor_distributions/loglogistic.py index 57e8cd0..ddc9c41 100644 --- a/distributions/loglogistic.py +++ b/pytensor_distributions/loglogistic.py @@ -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): diff --git a/distributions/lognormal.py b/pytensor_distributions/lognormal.py similarity index 97% rename from distributions/lognormal.py rename to pytensor_distributions/lognormal.py index 6c3fb08..1227fa2 100644 --- a/distributions/lognormal.py +++ b/pytensor_distributions/lognormal.py @@ -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): diff --git a/distributions/moyal.py b/pytensor_distributions/moyal.py similarity index 96% rename from distributions/moyal.py rename to pytensor_distributions/moyal.py index 627d415..d3b3aeb 100644 --- a/distributions/moyal.py +++ b/pytensor_distributions/moyal.py @@ -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): diff --git a/distributions/multinomial.py b/pytensor_distributions/multinomial.py similarity index 100% rename from distributions/multinomial.py rename to pytensor_distributions/multinomial.py diff --git a/distributions/mvnormal.py b/pytensor_distributions/mvnormal.py similarity index 100% rename from distributions/mvnormal.py rename to pytensor_distributions/mvnormal.py diff --git a/distributions/mvstudentt.py b/pytensor_distributions/mvstudentt.py similarity index 97% rename from distributions/mvstudentt.py rename to pytensor_distributions/mvstudentt.py index 1a97e49..fe1eca3 100644 --- a/distributions/mvstudentt.py +++ b/pytensor_distributions/mvstudentt.py @@ -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): diff --git a/distributions/negativebinomial.py b/pytensor_distributions/negativebinomial.py similarity index 91% rename from distributions/negativebinomial.py rename to pytensor_distributions/negativebinomial.py index 79a73f5..cbe23b5 100644 --- a/distributions/negativebinomial.py +++ b/pytensor_distributions/negativebinomial.py @@ -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): diff --git a/distributions/normal.py b/pytensor_distributions/normal.py similarity index 97% rename from distributions/normal.py rename to pytensor_distributions/normal.py index e1c3946..9b9e5ed 100644 --- a/distributions/normal.py +++ b/pytensor_distributions/normal.py @@ -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): diff --git a/distributions/optimization.py b/pytensor_distributions/optimization.py similarity index 98% rename from distributions/optimization.py rename to pytensor_distributions/optimization.py index c5df102..c610714 100644 --- a/distributions/optimization.py +++ b/pytensor_distributions/optimization.py @@ -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): diff --git a/distributions/pareto.py b/pytensor_distributions/pareto.py similarity index 96% rename from distributions/pareto.py rename to pytensor_distributions/pareto.py index 9bcbdee..0ac19ac 100644 --- a/distributions/pareto.py +++ b/pytensor_distributions/pareto.py @@ -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): diff --git a/distributions/poisson.py b/pytensor_distributions/poisson.py similarity index 90% rename from distributions/poisson.py rename to pytensor_distributions/poisson.py index ceec3e0..d61ccfe 100644 --- a/distributions/poisson.py +++ b/pytensor_distributions/poisson.py @@ -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): diff --git a/distributions/rice.py b/pytensor_distributions/rice.py similarity index 96% rename from distributions/rice.py rename to pytensor_distributions/rice.py index c9648c4..63016bd 100644 --- a/distributions/rice.py +++ b/pytensor_distributions/rice.py @@ -1,6 +1,6 @@ import pytensor.tensor as pt -from distributions.helper import ( +from pytensor_distributions.helper import ( cdf_bounds, continuous_entropy, continuous_kurtosis, @@ -8,7 +8,7 @@ continuous_skewness, marcum_q1_complement, ) -from distributions.optimization import find_ppf +from pytensor_distributions.optimization import find_ppf def _laguerre_half(q): diff --git a/distributions/scaledinversechisquared.py b/pytensor_distributions/scaledinversechisquared.py similarity index 96% rename from distributions/scaledinversechisquared.py rename to pytensor_distributions/scaledinversechisquared.py index 9362ef8..096b9e8 100644 --- a/distributions/scaledinversechisquared.py +++ b/pytensor_distributions/scaledinversechisquared.py @@ -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): diff --git a/distributions/skew_studentt.py b/pytensor_distributions/skew_studentt.py similarity index 98% rename from distributions/skew_studentt.py rename to pytensor_distributions/skew_studentt.py index 94ea5b4..6452f1b 100644 --- a/distributions/skew_studentt.py +++ b/pytensor_distributions/skew_studentt.py @@ -2,7 +2,7 @@ from pytensor.tensor.math import betaincinv from pytensor.tensor.special import betaln -from distributions.helper import continuous_entropy, continuous_mode, ppf_bounds_cont +from pytensor_distributions.helper import continuous_entropy, continuous_mode, ppf_bounds_cont def _raw_moment(n, a, b): diff --git a/distributions/skewnormal.py b/pytensor_distributions/skewnormal.py similarity index 93% rename from distributions/skewnormal.py rename to pytensor_distributions/skewnormal.py index b9c854c..ff3d822 100644 --- a/distributions/skewnormal.py +++ b/pytensor_distributions/skewnormal.py @@ -1,8 +1,8 @@ import pytensor.tensor as pt -from distributions.halfnormal import entropy as halfnormal_entropy -from distributions.normal import entropy as normal_entropy -from distributions.optimization import find_ppf +from pytensor_distributions.halfnormal import entropy as halfnormal_entropy +from pytensor_distributions.normal import entropy as normal_entropy +from pytensor_distributions.optimization import find_ppf def mean(mu, sigma, alpha): diff --git a/distributions/studentt.py b/pytensor_distributions/studentt.py similarity index 92% rename from distributions/studentt.py rename to pytensor_distributions/studentt.py index 4a4df5f..478174f 100644 --- a/distributions/studentt.py +++ b/pytensor_distributions/studentt.py @@ -2,10 +2,10 @@ from pytensor.tensor.math import betaincinv from pytensor.tensor.special import betaln -from distributions.helper import ppf_bounds_cont -from distributions.normal import cdf as normal_logcdf -from distributions.normal import entropy as normal_entropy -from distributions.normal import logcdf as normal_logpdf +from pytensor_distributions.helper import ppf_bounds_cont +from pytensor_distributions.normal import cdf as normal_logcdf +from pytensor_distributions.normal import entropy as normal_entropy +from pytensor_distributions.normal import logcdf as normal_logpdf def mean(nu, mu, sigma): diff --git a/distributions/triangular.py b/pytensor_distributions/triangular.py similarity index 100% rename from distributions/triangular.py rename to pytensor_distributions/triangular.py diff --git a/distributions/truncatednormal.py b/pytensor_distributions/truncatednormal.py similarity index 97% rename from distributions/truncatednormal.py rename to pytensor_distributions/truncatednormal.py index 4ea83e3..8949190 100644 --- a/distributions/truncatednormal.py +++ b/pytensor_distributions/truncatednormal.py @@ -1,7 +1,7 @@ import pytensor.tensor as pt -from distributions import normal as Normal -from distributions.helper import cdf_bounds, logdiffexp, ppf_bounds_cont +from pytensor_distributions import normal as Normal +from pytensor_distributions.helper import cdf_bounds, logdiffexp, ppf_bounds_cont def _phi(z): diff --git a/distributions/uniform.py b/pytensor_distributions/uniform.py similarity index 96% rename from distributions/uniform.py rename to pytensor_distributions/uniform.py index 481ff4e..ec3fc17 100644 --- a/distributions/uniform.py +++ b/pytensor_distributions/uniform.py @@ -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(lower, upper): diff --git a/distributions/vonmises.py b/pytensor_distributions/vonmises.py similarity index 92% rename from distributions/vonmises.py rename to pytensor_distributions/vonmises.py index 9cb665b..c662479 100644 --- a/distributions/vonmises.py +++ b/pytensor_distributions/vonmises.py @@ -1,7 +1,7 @@ import pytensor.tensor as pt -from distributions.helper import von_mises_cdf -from distributions.optimization import von_mises_ppf +from pytensor_distributions.helper import von_mises_cdf +from pytensor_distributions.optimization import von_mises_ppf def mean(mu, kappa): diff --git a/distributions/wald.py b/pytensor_distributions/wald.py similarity index 93% rename from distributions/wald.py rename to pytensor_distributions/wald.py index e89a806..1fbf111 100644 --- a/distributions/wald.py +++ b/pytensor_distributions/wald.py @@ -1,7 +1,7 @@ import pytensor.tensor as pt -from distributions.helper import cdf_bounds -from distributions.optimization import find_ppf +from pytensor_distributions.helper import cdf_bounds +from pytensor_distributions.optimization import find_ppf def mean(mu, lam): diff --git a/distributions/weibull.py b/pytensor_distributions/weibull.py similarity index 97% rename from distributions/weibull.py rename to pytensor_distributions/weibull.py index 4cb04cd..6548805 100644 --- a/distributions/weibull.py +++ b/pytensor_distributions/weibull.py @@ -2,7 +2,7 @@ from pytensor.tensor.special import gamma 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): diff --git a/distributions/wishart.py b/pytensor_distributions/wishart.py similarity index 100% rename from distributions/wishart.py rename to pytensor_distributions/wishart.py diff --git a/distributions/zi_binomial.py b/pytensor_distributions/zi_binomial.py similarity index 93% rename from distributions/zi_binomial.py rename to pytensor_distributions/zi_binomial.py index 81c6c15..5fe906e 100644 --- a/distributions/zi_binomial.py +++ b/pytensor_distributions/zi_binomial.py @@ -1,8 +1,8 @@ import pytensor.tensor as pt -from distributions import binomial as Binomial -from distributions.helper import cdf_bounds, discrete_entropy, zi_mode -from distributions.optimization import find_ppf_discrete +from pytensor_distributions import binomial as Binomial +from pytensor_distributions.helper import cdf_bounds, discrete_entropy, zi_mode +from pytensor_distributions.optimization import find_ppf_discrete def mean(psi, n, p): diff --git a/distributions/zi_negativebinomial.py b/pytensor_distributions/zi_negativebinomial.py similarity index 93% rename from distributions/zi_negativebinomial.py rename to pytensor_distributions/zi_negativebinomial.py index 31f678e..3f12591 100644 --- a/distributions/zi_negativebinomial.py +++ b/pytensor_distributions/zi_negativebinomial.py @@ -1,8 +1,8 @@ import pytensor.tensor as pt -from distributions import negativebinomial as NegativeBinomial -from distributions.helper import cdf_bounds, discrete_entropy, sf_bounds, zi_mode -from distributions.optimization import find_ppf_discrete +from pytensor_distributions import negativebinomial as NegativeBinomial +from pytensor_distributions.helper import cdf_bounds, discrete_entropy, sf_bounds, zi_mode +from pytensor_distributions.optimization import find_ppf_discrete def mean(psi, n, p): diff --git a/distributions/zi_poisson.py b/pytensor_distributions/zi_poisson.py similarity index 92% rename from distributions/zi_poisson.py rename to pytensor_distributions/zi_poisson.py index 2d607e0..fc63deb 100644 --- a/distributions/zi_poisson.py +++ b/pytensor_distributions/zi_poisson.py @@ -1,8 +1,8 @@ import pytensor.tensor as pt -from distributions import poisson as Poisson -from distributions.helper import cdf_bounds, discrete_entropy, sf_bounds, zi_mode -from distributions.optimization import find_ppf_discrete +from pytensor_distributions import poisson as Poisson +from pytensor_distributions.helper import cdf_bounds, discrete_entropy, sf_bounds, zi_mode +from pytensor_distributions.optimization import find_ppf_discrete def mean(psi, mu): diff --git a/tests/test_asymmetriclaplace.py b/tests/test_asymmetriclaplace.py index 0a91951..b450ed0 100644 --- a/tests/test_asymmetriclaplace.py +++ b/tests/test_asymmetriclaplace.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import asymmetriclaplace as AsymmetricLaplace +from pytensor_distributions import asymmetriclaplace as AsymmetricLaplace from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_bernoulli.py b/tests/test_bernoulli.py index b8d2c1c..330cb10 100644 --- a/tests/test_bernoulli.py +++ b/tests/test_bernoulli.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import bernoulli as Bernoulli +from pytensor_distributions import bernoulli as Bernoulli from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_beta.py b/tests/test_beta.py index cf687dc..2d28e1e 100644 --- a/tests/test_beta.py +++ b/tests/test_beta.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import beta as Beta +from pytensor_distributions import beta as Beta from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_betabinomial.py b/tests/test_betabinomial.py index 2ee5352..1260a4b 100644 --- a/tests/test_betabinomial.py +++ b/tests/test_betabinomial.py @@ -4,7 +4,7 @@ import pytest from scipy import stats -from distributions import betabinomial as BetaBinomial +from pytensor_distributions import betabinomial as BetaBinomial from tests.helper_scipy import run_distribution_tests diff --git a/tests/test_betascaled.py b/tests/test_betascaled.py index 3a37b81..298dc70 100644 --- a/tests/test_betascaled.py +++ b/tests/test_betascaled.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import betascaled as BetaScaled +from pytensor_distributions import betascaled as BetaScaled from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_binomial.py b/tests/test_binomial.py index 592888e..84395d5 100644 --- a/tests/test_binomial.py +++ b/tests/test_binomial.py @@ -4,7 +4,7 @@ import pytest from scipy import stats -from distributions import binomial as Binomial +from pytensor_distributions import binomial as Binomial from tests.helper_scipy import run_distribution_tests diff --git a/tests/test_categorical.py b/tests/test_categorical.py index 6710632..2751b2f 100644 --- a/tests/test_categorical.py +++ b/tests/test_categorical.py @@ -6,7 +6,7 @@ from numpy.testing import assert_allclose from scipy import stats -from distributions import categorical as Categorical +from pytensor_distributions import categorical as Categorical def make_params(p, dtype="float64"): diff --git a/tests/test_cauchy.py b/tests/test_cauchy.py index fb8fc9c..61d1237 100644 --- a/tests/test_cauchy.py +++ b/tests/test_cauchy.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import cauchy as Cauchy +from pytensor_distributions import cauchy as Cauchy from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_chisquared.py b/tests/test_chisquared.py index a842672..993fc24 100644 --- a/tests/test_chisquared.py +++ b/tests/test_chisquared.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import chisquared as ChiSquared +from pytensor_distributions import chisquared as ChiSquared from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_dirichlet.py b/tests/test_dirichlet.py index a99e982..4e3178c 100644 --- a/tests/test_dirichlet.py +++ b/tests/test_dirichlet.py @@ -6,7 +6,7 @@ from numpy.testing import assert_allclose from scipy.stats import dirichlet, kurtosis, skew -from distributions import dirichlet as Dirichlet +from pytensor_distributions import dirichlet as Dirichlet TEST_CASES = [ np.array([1.0, 1.0]), diff --git a/tests/test_dirichletmultinomial.py b/tests/test_dirichletmultinomial.py index dc8158e..d7e91bf 100644 --- a/tests/test_dirichletmultinomial.py +++ b/tests/test_dirichletmultinomial.py @@ -6,7 +6,7 @@ from numpy.testing import assert_allclose from scipy.stats import dirichlet_multinomial, kurtosis, skew -from distributions import dirichletmultinomial as DirichletMultinomial +from pytensor_distributions import dirichletmultinomial as DirichletMultinomial TEST_CASES = [ (10, np.array([1.0, 1.0])), @@ -142,7 +142,7 @@ def test_dirichletmultinomial_constraints(): @pytest.mark.parametrize("n, a", TEST_CASES) def test_dirichletmultinomial_skewness(n, a): - from distributions import dirichletmultinomial as DM + from pytensor_distributions import dirichletmultinomial as DM p_n = pt.constant(n) p_a = pt.constant(a) diff --git a/tests/test_discreteuniform.py b/tests/test_discreteuniform.py index a5be679..c907d95 100644 --- a/tests/test_discreteuniform.py +++ b/tests/test_discreteuniform.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import discreteuniform as DiscreteUniform +from pytensor_distributions import discreteuniform as DiscreteUniform from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_discreteweibull.py b/tests/test_discreteweibull.py index f5cd661..92e7243 100644 --- a/tests/test_discreteweibull.py +++ b/tests/test_discreteweibull.py @@ -2,7 +2,7 @@ import pytest -from distributions import discreteweibull as DWeibull +from pytensor_distributions import discreteweibull as DWeibull from tests.helper_empirical import run_empirical_tests from tests.helper_scipy import make_params diff --git a/tests/test_exgaussian.py b/tests/test_exgaussian.py index 7c48fb6..e5a6a97 100644 --- a/tests/test_exgaussian.py +++ b/tests/test_exgaussian.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import exgaussian as ExGaussian +from pytensor_distributions import exgaussian as ExGaussian from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_exponential.py b/tests/test_exponential.py index a4f3c51..ece95a8 100644 --- a/tests/test_exponential.py +++ b/tests/test_exponential.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import exponential as Exponential +from pytensor_distributions import exponential as Exponential from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_gamma.py b/tests/test_gamma.py index 3083003..228435f 100644 --- a/tests/test_gamma.py +++ b/tests/test_gamma.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import gamma as Gamma +from pytensor_distributions import gamma as Gamma from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_geometric.py b/tests/test_geometric.py index 7f3bf65..0d70e03 100644 --- a/tests/test_geometric.py +++ b/tests/test_geometric.py @@ -4,7 +4,7 @@ import pytest from scipy import stats -from distributions import geometric as Geometric +from pytensor_distributions import geometric as Geometric from tests.helper_scipy import run_distribution_tests diff --git a/tests/test_gumbel.py b/tests/test_gumbel.py index 1ce91e3..542d31c 100644 --- a/tests/test_gumbel.py +++ b/tests/test_gumbel.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import gumbel as Gumbel +from pytensor_distributions import gumbel as Gumbel from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_halfcauchy.py b/tests/test_halfcauchy.py index 5bdbc2c..3ffd962 100644 --- a/tests/test_halfcauchy.py +++ b/tests/test_halfcauchy.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import halfcauchy as HalfCauchy +from pytensor_distributions import halfcauchy as HalfCauchy from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_halfnormal.py b/tests/test_halfnormal.py index 20656a8..e3ef158 100644 --- a/tests/test_halfnormal.py +++ b/tests/test_halfnormal.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import halfnormal as HalfNormal +from pytensor_distributions import halfnormal as HalfNormal from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_halfstudentt.py b/tests/test_halfstudentt.py index e8c2a77..d77b990 100644 --- a/tests/test_halfstudentt.py +++ b/tests/test_halfstudentt.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import halfstudentt as HalfStudentT +from pytensor_distributions import halfstudentt as HalfStudentT from tests.helper_scipy import make_params, run_distribution_tests # HalfStudentT is not defined in scipy diff --git a/tests/test_helper.py b/tests/test_helper.py index a2ccdcd..d7318f1 100644 --- a/tests/test_helper.py +++ b/tests/test_helper.py @@ -4,7 +4,7 @@ import pytensor.tensor as pt from numpy.testing import assert_allclose -from distributions.helper import zi_mode +from pytensor_distributions.helper import zi_mode class TestZiMode: @@ -56,7 +56,7 @@ def mock_logpdf(x, psi, mu): def test_zi_mode_with_zi_poisson(self): """Test zi_mode with actual ZI-Poisson distribution.""" - from distributions import zi_poisson as ZIPoisson + from pytensor_distributions import zi_poisson as ZIPoisson # High zero inflation (low psi) - mode should be 0 psi_low = pt.constant(0.3) @@ -73,7 +73,7 @@ def test_zi_mode_with_zi_poisson(self): def test_zi_mode_with_zi_binomial(self): """Test zi_mode with actual ZI-Binomial distribution.""" - from distributions import zi_binomial as ZIBinomial + from pytensor_distributions import zi_binomial as ZIBinomial # High zero inflation (low psi) - mode should be 0 psi_low = pt.constant(0.2) @@ -90,7 +90,7 @@ def test_zi_mode_with_zi_binomial(self): def test_zi_mode_with_zi_negativebinomial(self): """Test zi_mode with actual ZI-Negative Binomial distribution.""" - from distributions import zi_negativebinomial as ZINegBinom + from pytensor_distributions import zi_negativebinomial as ZINegBinom # High zero inflation (low psi) - mode should be 0 psi_low = pt.constant(0.2) @@ -109,7 +109,7 @@ def test_zi_mode_with_zi_negativebinomial(self): def test_zi_mode_vectorized(self): """Test that zi_mode works with array inputs.""" - from distributions import zi_poisson as ZIPoisson + from pytensor_distributions import zi_poisson as ZIPoisson # Array of psi values psi_array = pt.constant(np.array([0.1, 0.5, 0.99])) @@ -126,7 +126,7 @@ def test_zi_mode_vectorized(self): def test_zi_mode_preserves_shape(self): """Test that zi_mode preserves input shape.""" - from distributions import zi_poisson as ZIPoisson + from pytensor_distributions import zi_poisson as ZIPoisson psi = pt.constant(np.array([[0.9, 0.9], [0.9, 0.9]])) mu = pt.constant(np.array([[3.0, 5.0], [7.0, 10.0]])) diff --git a/tests/test_hypergeometric.py b/tests/test_hypergeometric.py index 9b91304..6da6a5c 100644 --- a/tests/test_hypergeometric.py +++ b/tests/test_hypergeometric.py @@ -2,7 +2,7 @@ import pytest from scipy import stats -from distributions import hypergeometric as Hypergeometric +from pytensor_distributions import hypergeometric as Hypergeometric from tests.helper_scipy import run_distribution_tests diff --git a/tests/test_inversegamma.py b/tests/test_inversegamma.py index 5a4f5fc..914c585 100644 --- a/tests/test_inversegamma.py +++ b/tests/test_inversegamma.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import inversegamma as InverseGamma +from pytensor_distributions import inversegamma as InverseGamma from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_kumaraswamy.py b/tests/test_kumaraswamy.py index 60790f0..44cdccd 100644 --- a/tests/test_kumaraswamy.py +++ b/tests/test_kumaraswamy.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import kumaraswamy as Kumaraswamy +from pytensor_distributions import kumaraswamy as Kumaraswamy from tests.helper_empirical import run_empirical_tests from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_laplace.py b/tests/test_laplace.py index 6cae018..0d8e042 100644 --- a/tests/test_laplace.py +++ b/tests/test_laplace.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import laplace as Laplace +from pytensor_distributions import laplace as Laplace from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_logistic.py b/tests/test_logistic.py index 84567b1..cf4778c 100644 --- a/tests/test_logistic.py +++ b/tests/test_logistic.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import logistic as Logistic +from pytensor_distributions import logistic as Logistic from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_logitnormal.py b/tests/test_logitnormal.py index be3045c..b9ff9a1 100644 --- a/tests/test_logitnormal.py +++ b/tests/test_logitnormal.py @@ -2,7 +2,7 @@ import pytest -from distributions import logitnormal as LogitNormal +from pytensor_distributions import logitnormal as LogitNormal from tests.helper_empirical import run_empirical_tests from tests.helper_scipy import make_params diff --git a/tests/test_loglogistic.py b/tests/test_loglogistic.py index 862dba8..87d074e 100644 --- a/tests/test_loglogistic.py +++ b/tests/test_loglogistic.py @@ -1,7 +1,7 @@ import pytest from scipy import stats -from distributions import loglogistic as LogLogistic +from pytensor_distributions import loglogistic as LogLogistic from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_lognormal.py b/tests/test_lognormal.py index 52926f4..4ae83c6 100644 --- a/tests/test_lognormal.py +++ b/tests/test_lognormal.py @@ -4,7 +4,7 @@ import pytest from scipy import stats -from distributions import lognormal as LogNormal +from pytensor_distributions import lognormal as LogNormal from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_moyal.py b/tests/test_moyal.py index cf68fc6..022a0c8 100644 --- a/tests/test_moyal.py +++ b/tests/test_moyal.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import moyal as Moyal +from pytensor_distributions import moyal as Moyal from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_multinomial.py b/tests/test_multinomial.py index fe7e7ce..aabd7b6 100644 --- a/tests/test_multinomial.py +++ b/tests/test_multinomial.py @@ -6,7 +6,7 @@ from numpy.testing import assert_allclose from scipy.stats import multinomial as scipy_multinomial -from distributions import multinomial as Multinomial +from pytensor_distributions import multinomial as Multinomial TEST_CASES = [ (10, np.array([0.5, 0.5])), diff --git a/tests/test_mvnormal.py b/tests/test_mvnormal.py index 88c6aa4..fac98d4 100644 --- a/tests/test_mvnormal.py +++ b/tests/test_mvnormal.py @@ -6,7 +6,7 @@ from numpy.testing import assert_allclose from scipy.stats import multivariate_normal -from distributions import mvnormal as MvNormal +from pytensor_distributions import mvnormal as MvNormal # Test parameters for multivariate normal TEST_CASES = [ diff --git a/tests/test_mvstudentt.py b/tests/test_mvstudentt.py index f9f7387..87b0cf8 100644 --- a/tests/test_mvstudentt.py +++ b/tests/test_mvstudentt.py @@ -6,7 +6,7 @@ from numpy.testing import assert_allclose from scipy.stats import multivariate_t -from distributions import mvstudentt as MvStudentT +from pytensor_distributions import mvstudentt as MvStudentT TEST_CASES = [ (5.0, np.array([0.0, 0.0]), np.eye(2)), diff --git a/tests/test_negativebinomial.py b/tests/test_negativebinomial.py index 58b7e05..424287a 100644 --- a/tests/test_negativebinomial.py +++ b/tests/test_negativebinomial.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import negativebinomial as NegativeBinomial +from pytensor_distributions import negativebinomial as NegativeBinomial from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_normal.py b/tests/test_normal.py index 82446bd..7da0531 100644 --- a/tests/test_normal.py +++ b/tests/test_normal.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import normal as Normal +from pytensor_distributions import normal as Normal from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_pareto.py b/tests/test_pareto.py index 5ca52b4..a073395 100644 --- a/tests/test_pareto.py +++ b/tests/test_pareto.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import pareto as Pareto +from pytensor_distributions import pareto as Pareto from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_poisson.py b/tests/test_poisson.py index ffe661b..a12d8fc 100644 --- a/tests/test_poisson.py +++ b/tests/test_poisson.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import poisson as Poisson +from pytensor_distributions import poisson as Poisson from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_rice.py b/tests/test_rice.py index 25f5816..16ae97c 100644 --- a/tests/test_rice.py +++ b/tests/test_rice.py @@ -3,7 +3,7 @@ from numpy.testing import assert_allclose from scipy import stats -from distributions import rice as Rice +from pytensor_distributions import rice as Rice from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_scaledinversechisquared.py b/tests/test_scaledinversechisquared.py index 25a7c72..6adbb56 100644 --- a/tests/test_scaledinversechisquared.py +++ b/tests/test_scaledinversechisquared.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import scaledinversechisquared as ScaledInverseChiSquared +from pytensor_distributions import scaledinversechisquared as ScaledInverseChiSquared from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_skew_studentt.py b/tests/test_skew_studentt.py index 8e72af2..8b14c51 100644 --- a/tests/test_skew_studentt.py +++ b/tests/test_skew_studentt.py @@ -4,7 +4,7 @@ import pytest from scipy.stats import jf_skew_t -from distributions import skew_studentt as SkewStudentT +from pytensor_distributions import skew_studentt as SkewStudentT from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_skewnormal.py b/tests/test_skewnormal.py index f16ebc2..1cefc9a 100644 --- a/tests/test_skewnormal.py +++ b/tests/test_skewnormal.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import skewnormal as SkewNormal +from pytensor_distributions import skewnormal as SkewNormal from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_studentt.py b/tests/test_studentt.py index c1a4ae0..9a317d2 100644 --- a/tests/test_studentt.py +++ b/tests/test_studentt.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import studentt as StudentT +from pytensor_distributions import studentt as StudentT from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_triangular.py b/tests/test_triangular.py index 8849886..1ea974c 100644 --- a/tests/test_triangular.py +++ b/tests/test_triangular.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import triangular as Triangular +from pytensor_distributions import triangular as Triangular from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_truncatednormal.py b/tests/test_truncatednormal.py index 2d19000..5071d56 100644 --- a/tests/test_truncatednormal.py +++ b/tests/test_truncatednormal.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import truncatednormal as TruncatedNormal +from pytensor_distributions import truncatednormal as TruncatedNormal from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_uniform.py b/tests/test_uniform.py index 85acb38..fdc3297 100644 --- a/tests/test_uniform.py +++ b/tests/test_uniform.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import uniform as Uniform +from pytensor_distributions import uniform as Uniform from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_vonmises.py b/tests/test_vonmises.py index b4636a4..f1a7819 100644 --- a/tests/test_vonmises.py +++ b/tests/test_vonmises.py @@ -4,7 +4,7 @@ import pytest from scipy import stats -from distributions import vonmises as VonMises +from pytensor_distributions import vonmises as VonMises from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_wald.py b/tests/test_wald.py index e3bbcc5..f8cae36 100644 --- a/tests/test_wald.py +++ b/tests/test_wald.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import wald as Wald +from pytensor_distributions import wald as Wald from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_weibull.py b/tests/test_weibull.py index 1b81ca1..380940a 100644 --- a/tests/test_weibull.py +++ b/tests/test_weibull.py @@ -3,7 +3,7 @@ import pytest from scipy import stats -from distributions import weibull as Weibull +from pytensor_distributions import weibull as Weibull from tests.helper_scipy import make_params, run_distribution_tests diff --git a/tests/test_wishart.py b/tests/test_wishart.py index 7bf55f9..ab06b00 100644 --- a/tests/test_wishart.py +++ b/tests/test_wishart.py @@ -6,7 +6,7 @@ from numpy.testing import assert_allclose from scipy.stats import wishart as scipy_wishart -from distributions import wishart as Wishart +from pytensor_distributions import wishart as Wishart # Test cases: (nu, V) TEST_CASES = [ diff --git a/tests/test_zi_binomial.py b/tests/test_zi_binomial.py index ece9092..f6c4650 100644 --- a/tests/test_zi_binomial.py +++ b/tests/test_zi_binomial.py @@ -6,7 +6,7 @@ from numpy.testing import assert_allclose from scipy.stats import kurtosis, skew -from distributions import zi_binomial as ZIBinomial +from pytensor_distributions import zi_binomial as ZIBinomial def make_params(psi, n, p): @@ -152,7 +152,7 @@ def test_zi_binomial_mean_variance(params): def test_zi_binomial_reduces_to_binomial(): """Test that when psi=1, ZI-Binomial reduces to standard Binomial.""" - from distributions import binomial as Binomial + from pytensor_distributions import binomial as Binomial psi = 1.0 n = 10 diff --git a/tests/test_zi_negativebinomial.py b/tests/test_zi_negativebinomial.py index 0fcf672..c71b9fa 100644 --- a/tests/test_zi_negativebinomial.py +++ b/tests/test_zi_negativebinomial.py @@ -6,7 +6,7 @@ from numpy.testing import assert_allclose from scipy.stats import kurtosis, skew -from distributions import zi_negativebinomial as ZINegativeBinomial +from pytensor_distributions import zi_negativebinomial as ZINegativeBinomial def make_params(*values, dtype="float64"): @@ -149,7 +149,7 @@ def test_zi_negativebinomial_mean_variance(params): def test_zi_negativebinomial_reduces_to_negativebinomial(): """Test that when psi=1, ZI-NegBinom reduces to standard NegBinom.""" - from distributions import negativebinomial as NegativeBinomial + from pytensor_distributions import negativebinomial as NegativeBinomial psi = 1.0 n = 5.0 diff --git a/tests/test_zi_poisson.py b/tests/test_zi_poisson.py index 96a1cb1..e79e538 100644 --- a/tests/test_zi_poisson.py +++ b/tests/test_zi_poisson.py @@ -5,7 +5,7 @@ import pytest from numpy.testing import assert_allclose -from distributions import zi_poisson as ZIPoisson +from pytensor_distributions import zi_poisson as ZIPoisson from tests.helper_empirical import run_empirical_tests @@ -143,7 +143,7 @@ def test_zi_poisson_mean_variance(params): def test_zi_poisson_reduces_to_poisson(): """Test that when psi=1, ZI-Poisson reduces to standard Poisson.""" - from distributions import poisson as Poisson + from pytensor_distributions import poisson as Poisson psi = 1.0 mu = 3.0