fix

secretflow · Jan 7, 2025 · ba98dd6 · ba98dd6
1 parent 0491ab6
commit ba98dd6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git 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
@@ -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"],
 )