Fix units of N after norm

gyorilab · Jul 12, 2023 · bf859b7 · bf859b7
1 parent 58ee953
commit bf859b7
Show file tree

Hide file tree

Showing 3 changed files with 189 additions and 31 deletions.
diff --git a/mira/metamodel/ops.py b/mira/metamodel/ops.py
@@ -490,6 +490,8 @@ def counts_to_dimensionless(tm: TemplateModel,
         if p.units:
             (coeff, exponent) = \
                 p.units.expression.args[0].as_coeff_exponent(counts_unit_symbol)
+            if isinstance(exponent, sympy.core.numbers.One):
+                exponent = 1
             if exponent:
                 p.units.expression = \
                     SympyExprStr(p.units.expression.args[0] /