From 06c37c6442d058589d44fe2e076351279ec8fffc Mon Sep 17 00:00:00 2001 From: kinaryml Date: Wed, 15 Mar 2023 09:11:06 -0700 Subject: [PATCH] Updated mediapipe/python/BUILD and tests --- mediapipe/python/BUILD | 1 + mediapipe/tasks/python/test/vision/face_landmarker_test.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/mediapipe/python/BUILD b/mediapipe/python/BUILD index 8aa11f5f..879f8281 100644 --- a/mediapipe/python/BUILD +++ b/mediapipe/python/BUILD @@ -37,6 +37,7 @@ pybind_extension( deps = [ ":builtin_calculators", ":builtin_task_graphs", + "//mediapipe/tasks/cc/vision/face_geometry/calculators:geometry_pipeline_calculator", "//mediapipe/python/pybind:calculator_graph", "//mediapipe/python/pybind:image", "//mediapipe/python/pybind:image_frame", diff --git a/mediapipe/tasks/python/test/vision/face_landmarker_test.py b/mediapipe/tasks/python/test/vision/face_landmarker_test.py index eec19d58..fe189128 100644 --- a/mediapipe/tasks/python/test/vision/face_landmarker_test.py +++ b/mediapipe/tasks/python/test/vision/face_landmarker_test.py @@ -105,7 +105,7 @@ class ModelFileType(enum.Enum): FILE_NAME = 2 -class HandLandmarkerTest(parameterized.TestCase): +class FaceLandmarkerTest(parameterized.TestCase): def setUp(self): super().setUp()