Skip to content

Commit 6619550

Browse files
authored
Port Prior class from PyMC-Marketing (#470)
* port modules * port tests * remove references and bring back deserialize * add to requirements * add check for list * add pydantic * add pytest-mock * fix tests * add preliz * remove references to pymc-marketig * install test for tests * add to the docs * fix the typo * remove deserialize module * remove the deserialize docs
1 parent eab41fa commit 6619550

File tree

6 files changed

+2130
-3
lines changed

6 files changed

+2130
-3
lines changed

.github/workflows/test.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ jobs:
5353
cache-environment: true
5454
- name: Install pymc-extras
5555
run: |
56-
pip install -e .
56+
pip install -e ".[dev]"
5757
python --version
5858
- name: Run tests
5959
run: |
@@ -97,7 +97,7 @@ jobs:
9797
cache-environment: true
9898
- name: Install pymc-extras
9999
run: |
100-
pip install -e .
100+
pip install -e ".[dev]"
101101
python --version
102102
- name: Run tests
103103
# This job uses a cmd shell, therefore the environment variable syntax is different!

conda-envs/environment-test.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,8 @@ dependencies:
1313
- pytest
1414
- pytest-cov
1515
- libgcc<15
16+
- pydantic>=2.0.0
17+
- preliz
1618
- pip
1719
- pip:
1820
- jax

docs/api_reference.rst

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,21 @@ Distributions
4646
Skellam
4747
histogram_approximation
4848

49+
Prior
50+
=====
51+
52+
.. currentmodule:: pymc_extras.prior
53+
.. autosummary::
54+
:toctree: generated/
55+
56+
create_dim_handler
57+
handle_dims
58+
Prior
59+
VariableFactory
60+
sample_prior
61+
Censored
62+
Scaled
63+
4964

5065
Transforms
5166
==========

0 commit comments

Comments
 (0)