Fix depth condition bug when only depth condition is configured.

PiperOrigin-RevId: 567080598
This commit is contained in:
MediaPipe Team 2023-09-20 14:26:48 -07:00 committed by Copybara-Service
parent 41a012721f
commit 223641a73c

View File

@ -683,6 +683,7 @@ public final class ImageGenerator extends BaseVisionTaskApi {
.setEdgeConditionTypeOptions(edgeConditionOptions().get().convertToProto())
.build())
.build());
}
if (depthConditionOptions().isPresent()) {
taskOptionsBuilder.addControlPluginGraphsOptions(
ControlPluginGraphOptionsProto.ControlPluginGraphOptions.newBuilder()
@ -696,7 +697,6 @@ public final class ImageGenerator extends BaseVisionTaskApi {
.build())
.build());
}
}
return Any.newBuilder()
.setTypeUrl(
"type.googleapis.com/mediapipe.tasks.vision.image_generator.proto.ImageGeneratorGraphOptions")