From a22b999e98474403b03d4367781b480666aac668 Mon Sep 17 00:00:00 2001 From: Jyrki Launonen Date: Sun, 1 Sep 2024 21:20:46 +0300 Subject: [PATCH] Fix provision result type check At times, the result might be just an integer, such as when using min. --- kirppu/provision.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kirppu/provision.py b/kirppu/provision.py index e084e9b..255c1f4 100644 --- a/kirppu/provision.py +++ b/kirppu/provision.py @@ -65,7 +65,7 @@ def run_function(cls, provision_function, sold_and_compensated) -> Optional[Deci _r = run(provision_function, sold_and_compensated=sold_and_compensated) - assert _r is None or isinstance(_r, Decimal), "Value returned from function must be null or a number" + assert _r is None or isinstance(_r, (Decimal, int)), "Value returned from function must be null or a number" return _r def _run_function(self, items: Optional[QuerySet] = None) -> Optional[Decimal]: