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