Merge pull request #83 from nmdickson/hotfix/numdens-scaling

Hotfix/numdens scaling

Author: nmdickson

fitter/probabilities/probabilities.py

@@ -548,12 +548,15 @@ def likelihood_number_density(model, ndensity, *,
     model_Σ = model.Sigmaj[mass_bin] / model.mj[mass_bin]
 
     # Interpolated the model data at the measurement locations
-    interpolated = np.interp(obs_r, model_r, model_Σ)
+    interpolated = np.interp(obs_r, model_r, model_Σ).to(obs_Σ.unit)
 
     # Calculate K scaling factor
     K = (np.sum(obs_Σ * interpolated / yerr**2)
          / np.sum(interpolated**2 / yerr**2))
 
+    if np.isnan(K):
+        K = 1
+
     interpolated *= K
 
     return likelihood(obs_Σ, interpolated, yerr)