From dafe129a97d9b8a4989fff27d8aad3ea44884c4a Mon Sep 17 00:00:00 2001
From: Manos <manos@delkappa.com>
Date: Tue, 5 Oct 2021 05:32:48 +0200
Subject: [PATCH] expt_lmdk_sel: Again

---
 code/expt/expt_lmdk_sel.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/code/expt/expt_lmdk_sel.py b/code/expt/expt_lmdk_sel.py
index 7e84939..b05bde9 100644
--- a/code/expt/expt_lmdk_sel.py
+++ b/code/expt/expt_lmdk_sel.py
@@ -61,9 +61,10 @@ def main(args):
             # Turn part to histogram
             hist_pt, _ = lmdk_lib.get_hist(pt, lmdks_pt)
             # Get an option for this part
-            res_pt = opts[0]
             if len(opts) > 1:
               res_pt, _ = exp_mech.exponential(hist_pt, opts, exp_mech.score, 1.0, e)
+            elif len(opts) > 0:
+              res_pt = opts[0]
             # Merge options of all parts
             res[pt_i] = np.sum(res_pt)
           # Calculate MAE