| diff --git a/bazel/cython_library.bzl b/bazel/cython_library.bzl |
| index ed6ccc34ac..8618605df9 100644 |
| --- a/bazel/cython_library.bzl |
| +++ b/bazel/cython_library.bzl |
| @@ -74,7 +74,10 @@ def pyx_library(name, deps = [], py_deps = [], srcs = [], **kwargs): |
| native.cc_binary( |
| name = shared_object_name, |
| srcs = [stem + ".cpp"], |
| - deps = deps + ["@local_config_python//:python_headers"], |
| + deps = deps + [ |
| + "@rules_python//python/cc:current_py_cc_headers", |
| + "@rules_python//python/cc:current_py_cc_libs", |
| + ], |
| defines = defines, |
| linkshared = 1, |
| ) |
| diff --git a/bazel/python_rules.bzl b/bazel/python_rules.bzl |
| index e6cf84dca2..cf4be5b528 100644 |
| --- a/bazel/python_rules.bzl |
| +++ b/bazel/python_rules.bzl |
| @@ -209,11 +209,7 @@ def _generate_pb2_grpc_src_impl(context): |
| arguments = [] |
| tools = [context.executable._protoc, context.executable._grpc_plugin] |
| out_dir = get_out_dir(protos, context) |
| - if out_dir.import_path: |
| - # is virtual imports |
| - out_path = get_include_directory(out_files[0]) |
| - else: |
| - out_path = out_dir.path |
| + out_path = out_dir.path |
| arguments += get_plugin_args( |
| context.executable._grpc_plugin, |
| plugin_flags, |
