Fix regression in ElemMultExpr from introduction of index types

1 files changed, 9 insertions, 3 deletions

diff --git a/polymatrix/expression/mixins/elemmultexprmixin.py b/polymatrix/expression/mixins/elemmultexprmixin.py
index 0d89361..fe5d6ab 100644
--- a/polymatrix/expression/mixins/elemmultexprmixin.py
+++ b/polymatrix/expression/mixins/elemmultexprmixin.py
@@ -4,6 +4,7 @@ import typing
 import dataclassabc
 
 from polymatrix.polymatrix.init import init_poly_matrix, init_broadcast_poly_matrix
+from polymatrix.polymatrix.typing import MonomialIndex
 from polymatrix.expression.mixins.expressionbasemixin import ExpressionBaseMixin
 from polymatrix.polymatrix.mixins import PolyMatrixMixin
 from polymatrix.polymatrix.abc import PolyMatrix
@@ -60,9 +61,14 @@ class ElemMultExprMixin(ExpressionBaseMixin):
 
                     # monomial = tuple(sorted(left_monomial + right_monomial))
 
-                    new_monomial = merge_monomial_indices(
-                        (left_monomial, right_monomial)
-                    )
+                    # TODO: This line broke since the introduction of index
+                    # types, investigate why. Replacing it
+                    # MonomialIndex.product fixes the issue
+
+                    # new_monomial = merge_monomial_indices(
+                    #     (left_monomial, right_monomial)
+                    # )
+                    new_monomial = MonomialIndex.product(left_monomial, right_monomial)
 
                     if new_monomial not in polynomial:
                         polynomial[new_monomial] = 0