Skip to content

Commit 7a6263a

Browse files
committed
some additional ops to keep IW DOF int
1 parent ecd37be commit 7a6263a

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/carbs/extended_targets/ggiwMerge.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,12 +115,13 @@ def ggiw_merge(w:list, means:list, covs:list, alphas:list, betas:list, IWdof:lis
115115
bracket=[shape_d+2.0, 10000.0])
116116

117117
if solve_res.converged:
118+
118119
dof_merged = solve_res.root
119120
else:
120121
dof_merged = exp_dof
121122
else:
122123
dof_merged = exp_dof
123-
dof_merged = max(dof_merged, 2 * shape_d + 3) # Numerical hack to ensure IW is well defined
124+
dof_merged = float(round(max(dof_merged, 2 * shape_d + 3))) # Numerical hack to ensure IW is well defined
124125

125126
temp = sum(n * (a - shape_d - 1) * inv_sc for n,a,inv_sc in zip(w, IWdof, inv_scale))
126127
scale_merged = w_merged * (dof_merged - shape_d - 1) * np.linalg.inv(temp)

0 commit comments

Comments
 (0)