Skip to content

Commit baad3d9

Browse files
committed
Move pathfinder module to pymc_extras
- Move pathfinder module from pymc_experimental to pymc_extras - Update directory structure to match upstream repository
1 parent 382aeb7 commit baad3d9

File tree

6 files changed

+6
-6
lines changed

6 files changed

+6
-6
lines changed

pymc_experimental/inference/pathfinder/__init__.py

Lines changed: 0 additions & 3 deletions
This file was deleted.
Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
from pymc_extras.inference.pathfinder.pathfinder import fit_pathfinder
2+
3+
__all__ = ["fit_pathfinder"]

pymc_experimental/inference/pathfinder/pathfinder.py renamed to pymc_extras/inference/pathfinder/pathfinder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,10 +62,10 @@
6262
from rich.table import Table
6363
from rich.text import Text
6464

65-
from pymc_experimental.inference.pathfinder.importance_sampling import (
65+
from pymc_extras.inference.pathfinder.importance_sampling import (
6666
importance_sampling as _importance_sampling,
6767
)
68-
from pymc_experimental.inference.pathfinder.lbfgs import (
68+
from pymc_extras.inference.pathfinder.lbfgs import (
6969
LBFGS,
7070
LBFGSException,
7171
LBFGSInitFailed,

tests/test_pathfinder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def test_pathfinder(inference_backend):
6262
def test_bfgs_sample():
6363
import pytensor.tensor as pt
6464

65-
from pymc_experimental.inference.pathfinder.pathfinder import (
65+
from pymc_extras.inference.pathfinder.pathfinder import (
6666
alpha_recover,
6767
bfgs_sample,
6868
inverse_hessian_factors,

0 commit comments

Comments
 (0)