summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorNao Pross <np@0hm.ch>2024-05-26 19:30:18 +0200
committerNao Pross <np@0hm.ch>2024-05-27 00:02:26 +0200
commit2b24df74cd6179e86cbe0bbf0f9d02ef23b1f8bd (patch)
tree530acef11dbcf26d98edb054a07d5d6f9620f3b4
parentCreate poly.arange and ARangeExprMixin (diff)
downloadpolymatrix-2b24df74cd6179e86cbe0bbf0f9d02ef23b1f8bd.tar.gz
polymatrix-2b24df74cd6179e86cbe0bbf0f9d02ef23b1f8bd.zip
Fix bug in CombinationExprMixin
-rw-r--r--polymatrix/expression/mixins/combinationsexprmixin.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/polymatrix/expression/mixins/combinationsexprmixin.py b/polymatrix/expression/mixins/combinationsexprmixin.py
index fdac6f5..ff2d201 100644
--- a/polymatrix/expression/mixins/combinationsexprmixin.py
+++ b/polymatrix/expression/mixins/combinationsexprmixin.py
@@ -50,15 +50,15 @@ class CombinationsExprMixin(ExpressionBaseMixin):
# Scalars are OK
nrows, ncols = deg_pm.shape
if nrows == 1 and ncols == 1:
- degrees = (deg_pm.at(0, 0).constant(),)
+ degrees = (deg_pm.scalar().constant(),)
# Column vectors are OK
- elif nrows == 1:
+ elif ncols == 1:
degrees = (deg_pm.at(i, 0).constant()
for i in range(nrows))
# Row vectors are OK
- elif ncols == 1:
+ elif nrows == 1:
degrees = (deg_pm.at(0, i).constant()
for i in range(ncols))