expt_lmdk_sel: Dealing with corner cases

2021-10-05 05:16:29 +02:00
parent 840bd95c63
commit 438148edf7
1 changed files with 2 additions and 2 deletions
--- a/code/expt/expt_lmdk_sel.py
+++ b/code/expt/expt_lmdk_sel.py
@ -61,8 +61,8 @@ def main(args):
            # Turn part to histogram
            hist_pt, _ = lmdk_lib.get_hist(pt, lmdks_pt)
            # Get an option for this part
-            res_pt = np.array([])
-            if len(opts) > 0:
+            res_pt = opts[0]
+            if len(opts) > 1:
              res_pt, _ = exp_mech.exponential(hist_pt, opts, exp_mech.score, 1.0, e)
            # Merge options of all parts
            res[pt_i] = np.sum(res_pt)