Skip to content

Commit f12e0d2

Browse files
author
fabioferreira
committed
Port TensorFlow estimators/tests to PyTorch: Fix divergence measures and sims for CI
1 parent 9ff3aad commit f12e0d2

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

cde/model_fitting/divergences.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,9 +101,9 @@ def _div(x_tiled, y_samples):
101101
result = mc_integration_student_t(func, q.ndim_y, n_samples=n_samples, loc_proposal=mu_proposal, scale_proposal=std_proposal)
102102
result = np.asarray(result).reshape(-1)
103103
if n_measures == 1:
104-
distances[i] = result[0]
104+
distances[i] = result[0]
105105
else:
106-
distances[i] = result[:n_measures]
106+
distances[i] = result[:n_measures]
107107
assert distances.shape[0] == x_cond.shape[0]
108108
return distances
109109

0 commit comments

Comments
 (0)