allow compression exceptions

apache · Feb 4, 2025 · 4928480 · 4928480
1 parent e6b8ef0
commit 4928480
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/test/java/org/apache/sysds/test/component/compress/colgroup/CombineColGroups.java b/src/test/java/org/apache/sysds/test/component/compress/colgroup/CombineColGroups.java
@@ -31,6 +31,7 @@
 import org.apache.sysds.runtime.compress.CompressedMatrixBlock;
 import org.apache.sysds.runtime.compress.CompressionSettings;
 import org.apache.sysds.runtime.compress.CompressionSettingsBuilder;
+import org.apache.sysds.runtime.compress.DMLCompressionException;
 import org.apache.sysds.runtime.compress.colgroup.AColGroup;
 import org.apache.sysds.runtime.compress.colgroup.AColGroup.CompressionType;
 import org.apache.sysds.runtime.compress.colgroup.ColGroupFactory;
@@ -102,7 +103,7 @@ public void combine() {
 
 			TestUtils.compareMatricesBitAvgDistance(mb, ref, 0, 0, errMessage);
 		}
-		catch(NotImplementedException e) {
+		catch(NotImplementedException | DMLCompressionException e) {
 			// allowed
 		}
 		catch(Exception e) {