__version__ = "0.0.2"

from typing_extensions import Self

from .anns import Annotation, Category, Image

class COCO:
    def __init__(self: Self, annotation_path: str, image_folder_path: str) -> None: ...
    def get_anns(self: Self) -> list[Annotation]: ...
    def get_cats(self: Self) -> list[Category]: ...
    def get_imgs(self: Self) -> list[Image]: ...
    def get_img_anns(self: Self, img_id: int) -> list[Annotation]: ...
    def visualize_img(self: Self, img_id: int) -> None: ...
