From c51680740a77ae8ec693d6668150e2c6a2a1c2d3 Mon Sep 17 00:00:00 2001 From: Aaron Pham <29749331+aarnphm@users.noreply.github.com> Date: Wed, 7 Dec 2022 16:57:08 -0800 Subject: [PATCH] fix: podman health Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com> --- src/bentoml/_internal/container/podman.py | 2 +- src/bentoml/_internal/server/grpc/servicer/v1/__init__.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/src/bentoml/_internal/container/podman.py b/src/bentoml/_internal/container/podman.py index 06b20565db3..221a77e4fca 100644 --- a/src/bentoml/_internal/container/podman.py +++ b/src/bentoml/_internal/container/podman.py @@ -41,7 +41,7 @@ def health() -> bool: .strip("\n"") # strip quotation marks from JSON format. .lower() ) - return output == "running" + return output == '"running"' def parse_dict_opt(d: dict[str, str]) -> str: diff --git a/src/bentoml/_internal/server/grpc/servicer/v1/__init__.py b/src/bentoml/_internal/server/grpc/servicer/v1/__init__.py index 224f2b9878b..49ad48b0bcf 100644 --- a/src/bentoml/_internal/server/grpc/servicer/v1/__init__.py +++ b/src/bentoml/_internal/server/grpc/servicer/v1/__init__.py @@ -7,6 +7,7 @@ from typing import TYPE_CHECKING import anyio + from .....utils import LazyLoader from ......exceptions import InvalidArgument from ......exceptions import BentoMLException @@ -22,6 +23,7 @@ import grpc from google.protobuf import struct_pb2 + from ......grpc.v1 import service_pb2 as pb from ......grpc.v1 import service_pb2_grpc as services from ......grpc.types import BentoServicerContext