diff --git a/src/main/java/org/apache/sysds/runtime/frame/data/columns/HashIntegerArray.java b/src/main/java/org/apache/sysds/runtime/frame/data/columns/HashIntegerArray.java index 1b5a63e76b9..7900e7a8971 100644 --- a/src/main/java/org/apache/sysds/runtime/frame/data/columns/HashIntegerArray.java +++ b/src/main/java/org/apache/sysds/runtime/frame/data/columns/HashIntegerArray.java @@ -439,7 +439,7 @@ public boolean possiblyContainsNaN() { @Override protected long addValRecodeMap(Map map, long id, int i) { - Long val = getLong(i); + Integer val = Integer.valueOf(getInt(i)); Long v = map.putIfAbsent(val, id); if(v == null) id++; @@ -448,12 +448,12 @@ protected long addValRecodeMap(Map map, long id, int i) { @Override public void setM(Map map, AMapToData m, int i){ - m.set(i, map.get(getInt(i)).intValue() - 1); + m.set(i, map.get(Integer.valueOf(getInt(i))).intValue() - 1); } @Override public void setM(Map map, int si, AMapToData m, int i) { - final Integer v = getInt(i); + final Integer v = Integer.valueOf(getInt(i)); m.set(i, map.get(v).intValue() - 1); } diff --git a/src/main/java/org/apache/sysds/runtime/frame/data/columns/HashLongArray.java b/src/main/java/org/apache/sysds/runtime/frame/data/columns/HashLongArray.java index 85f237309bc..e5637aa37e0 100644 --- a/src/main/java/org/apache/sysds/runtime/frame/data/columns/HashLongArray.java +++ b/src/main/java/org/apache/sysds/runtime/frame/data/columns/HashLongArray.java @@ -436,7 +436,7 @@ public boolean possiblyContainsNaN() { @Override protected long addValRecodeMap(Map map, long id, int i) { - Integer val = getInt(i); + Long val = Long.valueOf(getLong(i)); Long v = map.putIfAbsent(val, id); if(v == null) id++; @@ -446,12 +446,12 @@ protected long addValRecodeMap(Map map, long id, int i) { @Override public void setM(Map map, AMapToData m, int i){ - m.set(i, map.get(getLong(i)).intValue() - 1); + m.set(i, map.get(Long.valueOf(getLong(i))).intValue() - 1); } @Override public void setM(Map map, int si, AMapToData m, int i) { - m.set(i, map.get(getLong(i)).intValue() - 1); + m.set(i, map.get(Long.valueOf(getLong(i))).intValue() - 1); } @Override