Project import generated by Copybara.

GitOrigin-RevId: 1e13be30e2c6838d4a2ff768a39c414bc80534bb

mediapipe/python/BUILD

@@ -12,10 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
+load("@org_tensorflow//tensorflow:tensorflow.bzl", "pybind_extension")
 
 licenses(["notice"])  # Apache 2.0
 
+package(default_visibility = ["//visibility:public"])
+
 pybind_extension(
     name = "_framework_bindings",
     srcs = ["framework_bindings.cc"],
@@ -31,8 +33,10 @@ pybind_extension(
             "-lopencv_imgcodecs",
         ],
     }),
+    module_name = "_framework_bindings",
     deps = [
         ":builtin_calculators",
+        ":builtin_task_graphs",
         "//mediapipe/python/pybind:calculator_graph",
         "//mediapipe/python/pybind:image",
         "//mediapipe/python/pybind:image_frame",
@@ -43,6 +47,7 @@ pybind_extension(
         "//mediapipe/python/pybind:resource_util",
         "//mediapipe/python/pybind:timestamp",
         "//mediapipe/python/pybind:validated_graph_config",
+        "//mediapipe/tasks/python/core/pybind:task_runner",
         # Type registration.
         "//mediapipe/framework:basic_types_registration",
         "//mediapipe/framework/formats:classification_registration",
@@ -76,3 +81,84 @@ cc_library(
         "//mediapipe/modules/selfie_segmentation:selfie_segmentation_cpu",
     ],
 )
+
+cc_library(
+    name = "builtin_task_graphs",
+    deps = [
+        "//mediapipe/tasks/cc/vision/object_detector:object_detector_graph",
+    ],
+)
+
+py_library(
+    name = "packet_creator",
+    srcs = ["packet_creator.py"],
+    srcs_version = "PY3",
+    deps = [
+        ":_framework_bindings",
+    ],
+)
+
+py_library(
+    name = "packet_getter",
+    srcs = ["packet_getter.py"],
+    srcs_version = "PY3",
+    deps = [
+        ":_framework_bindings",
+    ],
+)
+
+py_test(
+    name = "calculator_graph_test",
+    srcs = ["calculator_graph_test.py"],
+    python_version = "PY3",
+    srcs_version = "PY3",
+    deps = [
+        ":_framework_bindings",
+        ":packet_creator",
+        ":packet_getter",
+        "//mediapipe/framework:calculator_py_pb2",
+    ],
+)
+
+py_test(
+    name = "image_test",
+    srcs = ["image_test.py"],
+    python_version = "PY3",
+    srcs_version = "PY3",
+    deps = [
+        ":_framework_bindings",
+    ],
+)
+
+py_test(
+    name = "image_frame_test",
+    srcs = ["image_frame_test.py"],
+    python_version = "PY3",
+    srcs_version = "PY3",
+    deps = [
+        ":_framework_bindings",
+    ],
+)
+
+py_test(
+    name = "packet_test",
+    srcs = ["packet_test.py"],
+    python_version = "PY3",
+    srcs_version = "PY3",
+    deps = [
+        ":_framework_bindings",
+        ":packet_creator",
+        ":packet_getter",
+        "//mediapipe/framework/formats:detection_py_pb2",
+    ],
+)
+
+py_test(
+    name = "timestamp_test",
+    srcs = ["timestamp_test.py"],
+    python_version = "PY3",
+    srcs_version = "PY3",
+    deps = [
+        ":_framework_bindings",
+    ],
+)