commit d8c9495ae6d8e0507d113718b23dfd8fa5035e6d
author Sebastian <baunsgaard@tugraz.at> Wed Jun 03 18:57:30 2020 +0200
committer Matthias Boehm <mboehm7@gmail.com> Wed Jun 03 18:57:30 2020 +0200
tree 1bb97399827d76155f5011d88fc1a09fb85279e7
parent faec15810e18fc497e950b830811a919f5aa5c5f

[MINOR] Fix BuiltinFunctionExpression

Closes #937.
Closes #944.

src/main/java/org/apache/sysds/parser/BuiltinFunctionExpression.java

diff --git a/src/main/java/org/apache/sysds/parser/BuiltinFunctionExpression.java b/src/main/java/org/apache/sysds/parser/BuiltinFunctionExpression.java
index 96e2ebc..b358396 100644
--- a/src/main/java/org/apache/sysds/parser/BuiltinFunctionExpression.java
+++ b/src/main/java/org/apache/sysds/parser/BuiltinFunctionExpression.java
@@ -912,6 +912,14 @@
 		case NROW:
 		case NCOL:
 		case LENGTH:
+			checkNumParameters(1);
+			checkDataTypeParam(getFirstExpr(), 
+				DataType.FRAME, DataType.LIST, DataType.MATRIX);
+			output.setDataType(DataType.SCALAR);
+			output.setDimensions(0, 0);
+			output.setBlocksize(0);
+			output.setValueType(ValueType.INT64);
+			break;
 		case COUNT_DISTINCT:
 		case COUNT_DISTINCT_APPROX:
 			checkNumParameters(1);