Update PredictorUtils.java

SeldonIO · Jan 12, 2019 · c6570e7 · c6570e7
1 parent afeec4e
commit c6570e7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/engine/src/main/java/io/seldon/engine/predictors/PredictorUtils.java b/engine/src/main/java/io/seldon/engine/predictors/PredictorUtils.java
@@ -72,10 +72,10 @@ public static void add(DefaultData data, PrimitiveMatrix.DenseReceiver receiver)
             List<Integer> shapeList = data.getTensor().getShapeList();
 
             int rows = shapeList.get(0);
-            int columns = shapeList.get(1);
+            int cols = shapeList.get(1);
 
-            for (int i = 0; i < rows * columns; i++) {
-                receiver.add(i / columns, i % columns, valuesList.get(i));
+            for (int i = 0; i < rows * cols; i++) {
+                receiver.add(i / cols, i % cols, valuesList.get(i));
             }
 
         } else if (data.getDataOneofCase() == DataOneofCase.NDARRAY) {