From 4942ce48980fbaaf0286eb825c1b193d5589263c Mon Sep 17 00:00:00 2001 From: Linus Heck Date: Sun, 24 Sep 2023 21:07:50 +0200 Subject: [PATCH] Fix modulo, add power --- probably/prism/translate.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/probably/prism/translate.py b/probably/prism/translate.py index 05b4f6c..9a2021b 100644 --- a/probably/prism/translate.py +++ b/probably/prism/translate.py @@ -147,7 +147,9 @@ def expression_prism(expr: Expr) -> str: elif expr.operator == Binop.TIMES: return f"({lhs}) * ({rhs})" elif expr.operator == Binop.MODULO: - return f"({lhs}) * ({rhs})" + return f"mod({lhs}, {rhs})" + elif expr.operator == Binop.POWER: + return f"pow({lhs}, {rhs})" raise PrismTranslatorException("Operator not implemented:", expr) elif isinstance(expr, SubstExpr): raise PrismTranslatorException(