Bug fix calibration recipe for unit fitting in partition calibration

legend-exp · Nov 18, 2024 · 9bd7a2d · 9bd7a2d
1 parent 143524e
commit 9bd7a2d
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/ext/LegendSpecFitsRecipesBaseExt.jl b/ext/LegendSpecFitsRecipesBaseExt.jl
@@ -962,9 +962,18 @@ end
     bottom_margin --> (0, :mm)
     if report.type == :cal
         additional_pts = if !isempty(additional_pts)
-            μ_cal = report.f_fit.(additional_pts.μ) .* report.e_unit
-            (x = additional_pts.μ, y = ustrip.(report.e_unit, additional_pts.peaks),
-                residuals_norm = (value.(μ_cal) .- additional_pts.peaks) ./ uncertainty.(μ_cal))
+            μ_cal = if unit(first(additional_pts.μ)) != NoUnits
+                report.f_fit.(ustrip.(report.e_unit, additional_pts.μ))
+            else
+                report.f_fit.(additional_pts.μ)
+            end
+            p_cal = if unit(first(additional_pts.μ)) != NoUnits
+                ustrip.(report.e_unit, additional_pts.peaks)
+            else
+                additional_pts.peaks
+            end
+            (x = μ_cal, y = p_cal,
+                residuals_norm = (value.(μ_cal) .- p_cal)./ uncertainty.(μ_cal))
         else
             NamedTuple()
         end