fix: added 'hits' as dose unit, catch asserts when trying to compute …

…min/median/max dose
ErikBjare · Jun 25, 2024 · c0800c4 · c0800c4
1 parent 4886db1
commit c0800c4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/qslang/dose.py b/qslang/dose.py
@@ -30,6 +30,8 @@
 ureg.define("serving = x")  # for now
 ureg.define("puff = x")  # for now
 ureg.define("puffs = x")  # for now
+ureg.define("hit = x")  # for now
+ureg.define("hits = x")  # for now
 
 ureg.define("B = 10**9 * x")  # for noting billions of CFU, for example
 

diff --git a/qslang/main.py b/qslang/main.py
@@ -451,7 +451,7 @@ def _print_daily_doses(
         print(
             f" - min/median/max dose: {min_dose.amount_with_unit}/{median_dose.amount_with_unit}/{max_dose.amount_with_unit}"
         )
-    except pint.errors.DimensionalityError:
+    except (pint.errors.DimensionalityError, AssertionError):
         logger.warning(
             "Couldn't compute min/median/max doses due to inconsistent units"
         )