From 97727ad73e63e923c061b340515ea9c9d4024ee7 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 28 Nov 2014 13:39:35 -0600
Subject: [PATCH] Fix distributor power handling

---
 pymbolic/mapper/distributor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pymbolic/mapper/distributor.py b/pymbolic/mapper/distributor.py
index 2f6f96a..8abf993 100644
--- a/pymbolic/mapper/distributor.py
+++ b/pymbolic/mapper/distributor.py
@@ -90,12 +90,12 @@ class DistributeMapper(IdentityMapper):
     def map_power(self, expr):
         from pymbolic.primitives import Sum
 
+        newbase = self.rec(expr.base)
         if isinstance(expr.base, Product):
             return self.rec(pymbolic.flattened_product(
                 child**expr.exponent for child in newbase))
 
         if isinstance(expr.exponent, int):
-            newbase = self.rec(expr.base)
             if isinstance(newbase, Sum):
                 return self.map_product(
                         pymbolic.flattened_product(
-- 
GitLab