Update sunkit_spex/models/physical/albedo.py

sunpy · Jan 22, 2025 · 737a7ff · 737a7ff
1 parent b9299a8
commit 737a7ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sunkit_spex/models/physical/albedo.py b/sunkit_spex/models/physical/albedo.py
@@ -195,7 +195,7 @@ def _calculate_albedo_matrix(energy_edges: tuple[float], theta: float, anisotrop
     return albedo_interp.T
 
 
-@quantity_input
+@u.quantity_input
 def get_albedo_matrix(energy_edges: Quantity[u.keV], theta: Quantity[u.deg], anisotropy=1):
     r"""
     Get albedo correction matrix.