diff --git a/covalent/_workflow/electron.py b/covalent/_workflow/electron.py index cfe45f485..854fb1248 100644 --- a/covalent/_workflow/electron.py +++ b/covalent/_workflow/electron.py @@ -154,6 +154,7 @@ def get_op_function( "-": operator.sub, "*": operator.mul, "/": operator.truediv, + "**": operator.pow, } def rename(op1: Any, op: str, op2: Any) -> Callable: @@ -242,6 +243,9 @@ def __truediv__(self, other): def __rtruediv__(self, other): return self.get_op_function(other, self, "/") + def __pow__(self, other): + return self.get_op_function(self, other, "**") + def __int__(self): return int()