bugfix: fixes potential nonmatching b0mask with bvalues above 0. (#108)

AthenaEPI · Jan 8, 2021 · 21c3efc · 21c3efc
1 parent dcb0916
commit 21c3efc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dmipy/signal_models/cylinder_models.py b/dmipy/signal_models/cylinder_models.py
@@ -118,7 +118,7 @@ def spherical_mean(self, acquisition_scheme, **kwargs):
         bvals_ = bvals[bval_indices_above0]
         E_mean_ = ((np.sqrt(np.pi) * erf(np.sqrt(bvals_ * lambda_par))) /
                    (2 * np.sqrt(bvals_ * lambda_par)))
-        E_mean[~acquisition_scheme.shell_b0_mask] = E_mean_
+        E_mean[bval_indices_above0] = E_mean_
         return E_mean