Skip to content

Commit

Permalink
fix: lazy check for stubs via path when install local wheels (#3180)
Browse files Browse the repository at this point in the history
  • Loading branch information
aarnphm committed Nov 3, 2022
1 parent 0e8b7b8 commit f238251
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions src/bentoml/_internal/bento/build_dev_bentoml_whl.py
Expand Up @@ -43,11 +43,7 @@ def build_bentoml_editable_wheel(target_path: str) -> None:
raise BentoMLException("Could not find bentoml module location.")
bentoml_path = Path(module_location)

try:
from importlib import import_module

_ = import_module("bentoml.grpc.v1alpha1.service_pb2")
except ModuleNotFoundError:
if not Path(module_location, "grpc", "v1alpha1", "service_pb2.py").exists():
raise ModuleNotFoundError(
f"Generated stubs are not found. Make sure to run '{bentoml_path.as_posix()}/scripts/generate_grpc_stubs.sh' beforehand to generate gRPC stubs."
) from None
Expand Down

0 comments on commit f238251

Please sign in to comment.