diff --git a/node_wrappers/mediapipe_vision/pose/position.py b/node_wrappers/mediapipe_vision/pose/position.py index 5b4788d..db68049 100644 --- a/node_wrappers/mediapipe_vision/pose/position.py +++ b/node_wrappers/mediapipe_vision/pose/position.py @@ -2,7 +2,7 @@ import logging from typing import List, Tuple, Any, Optional # Import the base class -from ..location_landmark.universal_landmark_nodes import LandmarkPositionBaseNode +from ....src.mediapipe_vision.common.position import LandmarkPositionBase # Import the specific landmark type for clarity and potential future use from ....src.mediapipe_vision.types import POSE_LANDMARKS # Import the landmark definitions @@ -11,7 +11,7 @@ from ....src.mediapipe_vision.landmark_definitions import POSE_LANDMARK_TOOLTIP logger = logging.getLogger(__name__) _category = "Realtime Nodes/MediaPipe Vision/Pose/PoseLandmark/Position" -class PoseLandmarkPositionNode(LandmarkPositionBaseNode): +class PoseLandmarkPositionNode(LandmarkPositionBase): """Extracts position lists (x, y, z, vis, pres) for a specific landmark index from POSE_LANDMARKS across the batch.""" CATEGORY = _category