Handle unclassified annotations in SparseImageServerCommand

See https://forum.image.sc/t/error-in-create-combined-training-image/37687
petebankhead · May 13, 2020 · eb27109 · eb27109
1 parent 7337e73
commit eb27109
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/qupath-gui-fx/src/main/java/qupath/lib/gui/commands/SparseImageServerCommand.java b/qupath-gui-fx/src/main/java/qupath/lib/gui/commands/SparseImageServerCommand.java
@@ -115,10 +115,11 @@ protected SparseImageServer call() throws Exception {
 	}
 
 	static Predicate<PathObject> createPredicate(PathClass pathClass, boolean rectanglesOnly) {
+		var pathClass2 = pathClass == PathClassFactory.getPathClassUnclassified() ? null : pathClass;
 		if (rectanglesOnly)
-			return (PathObject p) -> p.isAnnotation() && p.getPathClass() == pathClass && p.getROI() instanceof RectangleROI;
+			return (PathObject p) -> p.isAnnotation() && p.getPathClass() == pathClass2 && p.getROI() instanceof RectangleROI;
 		else
-			return (PathObject p) -> p.isAnnotation() && p.getPathClass() == pathClass;
+			return (PathObject p) -> p.isAnnotation() && p.getPathClass() == pathClass2;
 	}