fix: handle units

scipp · Sep 23, 2024 · a1d7079 · a1d7079
1 parent 00e6738
commit a1d7079
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/src/scippneutron/absorption/material.py b/src/scippneutron/absorption/material.py
@@ -11,14 +11,15 @@ class Material:
     effective_sample_number_density: sc.Variable
 
     def attenuation_coefficient(self, wavelength):
-        return (
-            self.effective_sample_number_density
-            * (
-                self.scattering_cross_section
-                + self.attenuation_cross_section
+        '''Computes marginal attenuation per distance for
+        the given neutron wavelength.'''
+        return self.effective_sample_number_density * (
+            self.scattering_cross_section
+            + (
+                self.attenuation_cross_section
                 * (
                     wavelength
                     / sc.scalar(1.7982, unit='angstrom').to(unit=wavelength.unit)
                 )
-            )
-        ).to(unit='1/cm')
+            ).to(unit=self.scatter_cross_section.unit)
+        )