fix

Baunsgaard · Oct 22, 2024 · 1f9c468 · 1f9c468
1 parent 4bafc26
commit 1f9c468
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/main/java/org/apache/sysds/runtime/frame/data/columns/DDCArray.java b/src/main/java/org/apache/sysds/runtime/frame/data/columns/DDCArray.java
@@ -438,19 +438,19 @@ else if (dict instanceof OptionalArray){
 			OptionalArray<T> opt = (OptionalArray<T>) dict;
 
 			if(opt._n.get(i)){
-				if(dict instanceof HashIntegerArray)
+				if(opt._a instanceof HashIntegerArray)
 					m.set(i, map.get(((HashIntegerArray)opt._a).getInt(this.map.getIndex(i))).intValue()-1);
-				else if (dict instanceof HashLongArray)
+				else if (opt._a instanceof HashLongArray)
 					m.set(i, map.get(((HashLongArray)opt._a).getLong(this.map.getIndex(i))).intValue()-1);
 				else 
-					super.setM(map, si, m, i);
+					m.set(i, map.get(opt._a.get(this.map.getIndex(i))).intValue()-1);
 			}
 			else {
 				m.set(i, si);
 			}
 		}
 		else 
-			super.setM(map, si, m, i);
+			m.set(i, map.get(dict.get(this.map.getIndex(i))).intValue()-1);
 	}
 
 	@Override