Skip to content

Commit 4e4eeea

Browse files
committed
fix PoissonUncertainties module: needs to return outputs
test passing
1 parent 2b57000 commit 4e4eeea

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

src/modacor/modules/base_modules/poisson_uncertainties.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,3 +46,4 @@ def calculate(self, data: DataBundle, **kwargs: Any):
4646

4747
# Add the variance to the data
4848
data["signal"].variances["Poisson"] = np.clip(signal, 1, None)
49+
return {"signal": data["signal"]}

src/modacor/tests/integration/test_pipeline_run.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,4 +60,4 @@ def test_actual_processstep(flat_data):
6060
for node in pipeline.get_ready():
6161
node.execute(data=flat_data)
6262
pipeline.done(node)
63-
assert data["signal"].variances["Poisson"].mean().astype(int) == 10
63+
assert node.produced_outputs["signal"].variances["Poisson"].mean().astype(int) == 100

0 commit comments

Comments
 (0)