Fix modulo, add power

Philipp15b · Sep 24, 2023 · 4942ce4 · 4942ce4
1 parent c6eb165
commit 4942ce4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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(