add kappa to data

SyneRBI · Jul 4, 2024 · dc6a478 · dc6a478
1 parent 2bd78f4
commit dc6a478
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/petric.py b/petric.py
@@ -143,7 +143,7 @@ def construct_RDP(penalty_strength, initial_image, kappa, max_scaling=1e-3):
     return prior
 
 
-Dataset = namedtuple('Dataset', ['acquired_data', 'additive_term', 'mult_factors', 'OSEM_image', 'prior'])
+Dataset = namedtuple('Dataset', ['acquired_data', 'additive_term', 'mult_factors', 'OSEM_image', 'prior', 'kappa'])
 
 
 def get_data(srcdir=".", outdir=OUTDIR, sirf_verbosity=0):
@@ -165,7 +165,7 @@ def get_data(srcdir=".", outdir=OUTDIR, sirf_verbosity=0):
         penalty_strength = 1 / 700 # default choice
     prior = construct_RDP(penalty_strength, OSEM_image, kappa)
 
-    return Dataset(acquired_data, additive_term, mult_factors, OSEM_image, prior)
+    return Dataset(acquired_data, additive_term, mult_factors, OSEM_image, prior, kappa)
 
 
 if SRCDIR.is_dir():