fixes #1835

AgnostiqHQ · Oct 18, 2023 · e5718c0 · e5718c0
1 parent 6c69c82
commit e5718c0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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()