fix: edit generated python code st relative import works

The issue dates back to protocolbuffers/protobuf#1491
oom-ai · Nov 23, 2021 · 31eb06f · 31eb06f
1 parent 56ccac0
commit 31eb06f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Makefile b/Makefile
@@ -26,6 +26,8 @@ codegen:
 		--python_out=oomd/codegen \
 		--grpc-python_out=oomd/codegen \
 		oomd/oomd.proto
+	sed -i '' "s|import status_pb2|from . import status_pb2|g" oomd/codegen/oomd_pb2.py
+	sed -i '' "s|import oomd_pb2|from . import oomd_pb2|g" oomd/codegen/oomd_pb2_grpc.py
 
 .PHONY: build
 build: featctl