[systemds] branch main updated: [MINOR] Fix missed error in last Compressed Transform commit

[baunsgaard]

This is an automated email from the ASF dual-hosted git repository.

baunsgaard pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/systemds.git


The following commit(s) were added to refs/heads/main by this push:
     new 5ab14a408a [MINOR] Fix missed error in last Compressed Transform commit
5ab14a408a is described below

commit 5ab14a408aefecb2fef0ce71610b87131a6311c7
Author: baunsgaard <baunsga...@tugraz.at>
AuthorDate: Fri Jan 27 16:09:10 2023 +0100

    [MINOR] Fix missed error in last Compressed Transform commit
---
 .../sysds/runtime/transform/encode/MultiColumnEncoder.java       | 9 +++++++++
 .../test/component/frame/transform/transformCompressed.java      | 8 ++++----
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git 
a/src/main/java/org/apache/sysds/runtime/transform/encode/MultiColumnEncoder.java
 
b/src/main/java/org/apache/sysds/runtime/transform/encode/MultiColumnEncoder.java
index 59c1a3d09f..34a2ba8a76 100644
--- 
a/src/main/java/org/apache/sysds/runtime/transform/encode/MultiColumnEncoder.java
+++ 
b/src/main/java/org/apache/sysds/runtime/transform/encode/MultiColumnEncoder.java
@@ -865,6 +865,15 @@ public class MultiColumnEncoder implements Encoder {
                return sum;
        }
 
+       public int getNumExtraCols(IndexRange ixRange) {
+               List<ColumnEncoderDummycode> dc = 
getColumnEncoders(ColumnEncoderDummycode.class).stream()
+                       .filter(dce -> 
ixRange.inColRange(dce._colID)).collect(Collectors.toList());
+               if(dc.isEmpty()) {
+                       return 0;
+               }
+               return 
dc.stream().map(ColumnEncoderDummycode::getDomainSize).mapToInt(i -> i).sum() - 
dc.size();
+       }
+
        public <T extends ColumnEncoder> boolean containsEncoderForID(int 
colID, Class<T> type) {
                return getColumnEncoders(type).stream().anyMatch(encoder -> 
encoder.getColID() == colID);
        }
diff --git 
a/src/test/java/org/apache/sysds/test/component/frame/transform/transformCompressed.java
 
b/src/test/java/org/apache/sysds/test/component/frame/transform/transformCompressed.java
index 343aaf05ec..08a15c9aa2 100644
--- 
a/src/test/java/org/apache/sysds/test/component/frame/transform/transformCompressed.java
+++ 
b/src/test/java/org/apache/sysds/test/component/frame/transform/transformCompressed.java
@@ -98,10 +98,10 @@ public class transformCompressed {
                        FrameBlock outNormalMD = 
encoderNormal.getMetaData(null);
                        
                        
-                       LOG.error(outNormal);
-                       LOG.error(outCompressed);
-                       LOG.error(outCompressedMD);
-                       LOG.error(outNormalMD);
+                       // LOG.error(outNormal);
+                       // LOG.error(outCompressed);
+                       // LOG.error(outCompressedMD);
+                       // LOG.error(outNormalMD);
 
                        TestUtils.compareMatrices(outNormal, outCompressed, 0, 
"Not Equal after apply");
                        TestUtils.compareFrames(outNormalMD, outCompressedMD, 
true);