Skip to content

Commit 8bd4f75

Browse files
committed
added nested sampler uncertainty to pybind and outputs
1 parent 92c6d67 commit 8bd4f75

File tree

4 files changed

+7
-0
lines changed

4 files changed

+7
-0
lines changed

RATapi/outputs.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,7 @@ class DreamOutput(RATResult):
149149
@dataclass
150150
class NestedSamplerOutput(RATResult):
151151
logZ: float
152+
logZErr: float
152153
nestSamples: np.ndarray
153154
postSamples: np.ndarray
154155

@@ -230,6 +231,7 @@ def make_results(
230231

231232
nested_sampler_output = NestedSamplerOutput(
232233
logZ=bayes_results.nestedSamplerOutput.logZ,
234+
logZErr=bayes_results.nestedSamplerOutput.logZErr,
233235
nestSamples=bayes_results.nestedSamplerOutput.nestSamples,
234236
postSamples=bayes_results.nestedSamplerOutput.postSamples,
235237
)

cpp/rat.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -315,6 +315,7 @@ struct ConfidenceIntervals
315315
struct NestedSamplerOutput
316316
{
317317
real_T logZ;
318+
real_T logZErr;
318319
py::array_t<real_T> nestSamples;
319320
py::array_t<real_T> postSamples;
320321
};
@@ -1243,6 +1244,7 @@ BayesResults bayesResultsFromStruct9T(const RAT::struct9_T results)
12431244
bayesResults.confidenceIntervals.mean = pyArrayFromRatArray2d(results.confidenceIntervals.mean);
12441245

12451246
bayesResults.nestedSamplerOutput.logZ = results.nestedSamplerOutput.LogZ;
1247+
bayesResults.nestedSamplerOutput.logZErr = results.nestedSamplerOutput.LogZErr;
12461248
bayesResults.nestedSamplerOutput.nestSamples = pyArrayFromRatArray2d(results.nestedSamplerOutput.nestSamples);
12471249
bayesResults.nestedSamplerOutput.postSamples = pyArrayFromRatArray2d(results.nestedSamplerOutput.postSamples);
12481250

@@ -1445,6 +1447,7 @@ PYBIND11_MODULE(rat_core, m) {
14451447
py::class_<NestedSamplerOutput>(m, "NestedSamplerOutput")
14461448
.def(py::init<>())
14471449
.def_readwrite("logZ", &NestedSamplerOutput::logZ)
1450+
.def_readwrite("logZErr", &NestedSamplerOutput::logZErr)
14481451
.def_readwrite("nestSamples", &NestedSamplerOutput::nestSamples)
14491452
.def_readwrite("postSamples", &NestedSamplerOutput::postSamples);
14501453

tests/conftest.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5639,6 +5639,7 @@ def dream_results():
56395639
),
56405640
nestedSamplerOutput=RATapi.outputs.NestedSamplerOutput(
56415641
logZ=0.0,
5642+
logZErr=0.0,
56425643
nestSamples=np.array([[0.0, 0.0]]),
56435644
postSamples=np.array([[0.0, 0.0]]),
56445645
),

tests/test_outputs.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,7 @@ def dream_str():
126126
"),\n"
127127
"nestedSamplerOutput = NestedSamplerOutput(\n"
128128
"\tlogZ = 0.0,\n"
129+
"\tlogZErr = 0.0,\n"
129130
"\tnestSamples = Data array: [1 x 2],\n"
130131
"\tpostSamples = Data array: [1 x 2],\n"
131132
"),\n"

0 commit comments

Comments
 (0)