Skip to content

Commit 78ff887

Browse files
OriolAbrilbrandonwillard
authored andcommitted
Port InferenceData conversion code
1 parent 3713936 commit 78ff887

File tree

7 files changed

+1345
-69
lines changed

7 files changed

+1345
-69
lines changed

pymc3/__init__.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,12 @@ def __set_compiler_flags():
4040

4141
from pymc3 import gp, ode, sampling
4242
from pymc3.aesaraf import *
43-
from pymc3.backends import load_trace, save_trace
43+
from pymc3.backends import (
44+
load_trace,
45+
predictions_to_inference_data,
46+
save_trace,
47+
to_inference_data,
48+
)
4449
from pymc3.backends.tracetab import *
4550
from pymc3.blocking import *
4651
from pymc3.data import *

pymc3/backends/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@
6060
Saved backends can be loaded using `arviz.from_netcdf`
6161
6262
"""
63+
from pymc3.backends.arviz import predictions_to_inference_data, to_inference_data
6364
from pymc3.backends.ndarray import (
6465
NDArray,
6566
load_trace,

0 commit comments

Comments
 (0)