From 06776d20a5b008fd7cc475747b09feb9e9760585 Mon Sep 17 00:00:00 2001
From: Richard Levasseur <rlevasseur@google.com>
Date: Mon, 20 Nov 2023 11:51:50 -0800
Subject: [PATCH] Add `load()` statements for builtin Python rules/providers

PiperOrigin-RevId: 584089414
---
 bazel/BUILD                                   | 3 +++
 bazel/py_proto_library.bzl                    | 1 +
 python/pb_unit_tests/pyproto_test_wrapper.bzl | 2 +-
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/bazel/BUILD b/bazel/BUILD
index a891d615a2..551200cdbf 100644
--- a/bazel/BUILD
+++ b/bazel/BUILD
@@ -26,6 +26,9 @@ py_binary(
 bzl_library(
     name = "py_proto_library_bzl",
     srcs = ["py_proto_library.bzl"],
+    deps = [
+        "@rules_python//python:py_info_bzl",
+    ],
 )
 
 bzl_library(
diff --git a/bazel/py_proto_library.bzl b/bazel/py_proto_library.bzl
index 08ff9a8f94..41c4f6c2ba 100644
--- a/bazel/py_proto_library.bzl
+++ b/bazel/py_proto_library.bzl
@@ -15,6 +15,7 @@ But it hasn't been deeply tested or reviewed, and upb should not be in the
 business of vending py_proto_library(), so we keep it private to upb.
 """
 
+load("@rules_python//python:py_info.bzl", "PyInfo")
 load("@bazel_skylib//lib:paths.bzl", "paths")
 
 # begin:github_only
diff --git a/python/pb_unit_tests/pyproto_test_wrapper.bzl b/python/pb_unit_tests/pyproto_test_wrapper.bzl
index bd5d0c0704..3ddf630295 100644
--- a/python/pb_unit_tests/pyproto_test_wrapper.bzl
+++ b/python/pb_unit_tests/pyproto_test_wrapper.bzl
@@ -25,7 +25,7 @@ def pyproto_test_wrapper(name, deps = []):
 
 # begin:google_only
 #
-# load("//third_party/bazel_rules/rules_python/python:py_test.bzl", "py_test")
+# load("@rules_python//python:py_test.bzl", "py_test")
 #
 # def pyproto_test_wrapper(name):
 #     src = name + "_wrapper.py"