From 5153bda2091597fec0d2c8ff72f3d41d50f33162 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Tue, 16 Jul 2024 22:34:42 -0400 Subject: [PATCH] fix command line arg --- .../src/http_server/endpoints/health_check_handler.rs | 3 +++ .../src/job/job_types/workflow/live_portrait/command_args.rs | 5 +++-- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/crates/service/job/inference_job/src/http_server/endpoints/health_check_handler.rs b/crates/service/job/inference_job/src/http_server/endpoints/health_check_handler.rs index 6301d2aa0..788626257 100644 --- a/crates/service/job/inference_job/src/http_server/endpoints/health_check_handler.rs +++ b/crates/service/job/inference_job/src/http_server/endpoints/health_check_handler.rs @@ -16,6 +16,9 @@ use crate::http_server::http_server_shared_state::HttpServerSharedState; #[derive(Serialize)] pub struct HealthCheckResponse { pub success: bool, + + /// The server goes unhealthy when the consecutive job failure count exceeds + /// a configured threshold. pub is_healthy: bool, pub consecutive_failure_count: u64, diff --git a/crates/service/job/inference_job/src/job/job_types/workflow/live_portrait/command_args.rs b/crates/service/job/inference_job/src/job/job_types/workflow/live_portrait/command_args.rs index 810bc6fa3..9aa819271 100644 --- a/crates/service/job/inference_job/src/job/job_types/workflow/live_portrait/command_args.rs +++ b/crates/service/job/inference_job/src/job/job_types/workflow/live_portrait/command_args.rs @@ -32,8 +32,9 @@ impl CommandArgs for LivePortraitCommandArgs<'_> { command.push_str(" --output-filename "); command.push_str(&path_to_string(self.output_file)); - command.push_str(" --input-is-image "); - command.push_str(bool_to_str(self.input_is_image)); + if self.input_is_image { + command.push_str(" --input-is-image "); + } command.push_str(" ");