detect_tools/upn/__init__.py

from . import models
from .builder import (
    ARCHITECTURES,
    BACKBONES,
    DECODERS,
    ENCODERS,
    POS_EMBEDDINGS,
    build_architecture,
    build_backbone,
    build_decoder,
    build_encoder,
    build_position_embedding,
)
from .inference_wrapper import UPNWrapper
from .models.architecture import *
from .models.backbone import *
from .models.decoder import *
from .models.encoder import *
from .models.module import *
from .models.utils import *

__all__ = [
    "BACKBONES",
    "POS_EMBEDDINGS",
    "ENCODERS",
    "DECODERS",
    "ARCHITECTURES",
    "build_backbone",
    "build_position_embedding",
    "build_encoder",
    "build_decoder",
    "build_architecture",
    "UPNWrapper",
]

__all__ += (
    models.module.__all__
    + models.utils.__all__
    + models.architecture.__all__
    + models.backbone.__all__
    + models.encoder.__all__
    + models.decoder.__all__
    + models.module.__all__
    + models.utils.__all__
)