kd_node.py

from typing import List, Optional


class KDNode:
    """
    Represents a node in a KD-Tree.

    Attributes:
        point (List[float]): The k-dimensional point stored in this node.
        left (Optional[KDNode]): The left subtree of this node.
        right (Optional[KDNode]): The right subtree of this node.
    """

    def __init__(
        self,
        point: List[float],
        left: Optional["KDNode"] = None,
        right: Optional["KDNode"] = None,
    ) -> None:
        """
        Initializes a KDNode with a point and optional left and right children.

        Args:
            point (List[float]): The k-dimensional point to be stored in this node.
            left (Optional[KDNode]): The left subtree of this node. Defaults to None.
            right (Optional[KDNode]): The right subtree of this node. Defaults to None.
        """
        self.point = point
        self.left = left
        self.right = right