#22 handle empty power_ci results

precimed · Mar 27, 2020 · f4230fa · f4230fa
1 parent ad9afd6
commit f4230fa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/precimed/mixer/figures.py b/precimed/mixer/figures.py
@@ -229,10 +229,10 @@ def make_power_plot(data_vec, colors=None, traits=None, power_thresh=None):
 
         if 'power_ci' in data:
             if power_thresh is not None:
-                future_n_ci = [np.power(10, float(interp1d(data_power['svec'], np.log10(data_power['nvec']))(power_thresh))) for data_power in data['power_ci']]
+                future_n_ci = [np.power(10, float(interp1d(data_power['svec'], np.log10(data_power['nvec']))(power_thresh))) for data_power in data['power_ci'] if data_power]
                 leg_labels.append('{} {} ({})'.format(trait, display_auto(future_n_val), display_auto(np.std(future_n_ci))))
             else:
-                current_s_ci = [float(interp1d(np.log10(data_power['nvec']),data_power['svec'])(np.log10(data['options']['trait1_nval']))) for data_power in data['power_ci']]
+                current_s_ci = [float(interp1d(np.log10(data_power['nvec']),data_power['svec'])(np.log10(data['options']['trait1_nval']))) for data_power in data['power_ci'] if data_power]
                 leg_labels.append('{} {:.1f}% ({:.1f}%)'.format(trait, 100 * cs, 100*np.std(current_s_ci)))
         else:
             if power_thresh is not None: