From ba98dd6bca6e5bd77126b5c00a141f70fcd9de9e Mon Sep 17 00:00:00 2001
From: w-gc <25614556+w-gc@users.noreply.github.com>
Date: Tue, 7 Jan 2025 21:44:35 +0800
Subject: [PATCH] fix

---
 MODULE.bazel          |  1 +
 spu/utils/BUILD.bazel | 11 +++++------
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/MODULE.bazel b/MODULE.bazel
index 78f217f2..cf587849 100644
--- a/MODULE.bazel
+++ b/MODULE.bazel
@@ -56,6 +56,7 @@ bazel_dep(name = "platforms", version = "0.0.8")
 bazel_dep(name = "pybind11_bazel", version = "2.13.6")
 bazel_dep(name = "rules_python", version = "0.29.0")
 bazel_dep(name = "rules_proto", version = "6.0.0-rc1")
+bazel_dep(name = "rules_proto_grpc_python", version = "5.0.0")
 bazel_dep(name = "spdlog", version = "1.14.1")
 bazel_dep(name = "fmt", version = "11.0.2")
 bazel_dep(name = "abseil-cpp", version = "20240722.0")
diff --git a/spu/utils/BUILD.bazel b/spu/utils/BUILD.bazel
index f537897a..d6de5dc3 100644
--- a/spu/utils/BUILD.bazel
+++ b/spu/utils/BUILD.bazel
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-load("@grpc//bazel:python_rules.bzl", "py_grpc_library", "py_proto_library")
+load("@rules_proto_grpc_python//:defs.bzl", "python_proto_library", "python_grpc_library")
 load("@rules_proto//proto:defs.bzl", "proto_library")
 load("//bazel:spu.bzl", "spu_py_library")
 
@@ -32,15 +32,14 @@ proto_library(
     srcs = ["distributed.proto"],
 )
 
-py_proto_library(
+python_proto_library(
     name = "distributed_py_proto",
-    deps = [":distributed_proto"],
+    protos = [":distributed_proto"],
 )
 
-py_grpc_library(
+python_grpc_library(
     name = "distributed_py_proto_grpc",
-    srcs = [":distributed_proto"],
-    strip_prefixes = ["../.."],
+    protos = [":distributed_proto"],
     deps = [":distributed_py_proto"],
 )