Load Python rules from rules_python

PiperOrigin-RevId: 553849378
tensorflow · Aug 11, 2023 · 175b2eb · 175b2eb
1 parent 630af93
commit 175b2eb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/plugin/tensorboard_plugin_profile/convert/BUILD b/plugin/tensorboard_plugin_profile/convert/BUILD
@@ -3,6 +3,8 @@ load("@python_deps//:requirements.bzl", "requirement")
 # Description:
 #   Converter from protobuf to gviz/json formats.
 
+load("@rules_python/python:defs.bzl", "py_library", "py_test")
+
 visibility = ["//plugin:internal"]
 
 package(

diff --git a/plugin/tensorboard_plugin_profile/demo/BUILD b/plugin/tensorboard_plugin_profile/demo/BUILD
@@ -2,6 +2,9 @@ load("@python_deps//:requirements.bzl", "requirement")
 
 # Description:
 #  Demos for profiler plugin.
+
+load("@rules_python/python:defs.bzl", "py_binary")
+
 package(
     licenses = ["notice"],  # Apache 2.0
 )