Merge 5946bc0 into d0d4237

tBuLi · Apr 13, 2019 · 12f0220 · 12f0220
2 parents d0d4237 + 5946bc0
commit 12f0220
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/symfit/core/printing.py b/symfit/core/printing.py
@@ -73,3 +73,10 @@ def _print_Idx(self, expr):
         Print ``Idx`` objects.
         """
         return "{0}".format(self._print(expr.args[0]))
+
+    def _print_MatPow(self, expr):
+        if expr.shape == (1, 1):
+            # Scalar, so we can take a normal power.
+            return self._print_Pow(expr)
+        else:
+            return super(SymfitNumPyPrinter, self)._print_MatPow(expr)