from pathlib import Path
from dataclasses import dataclass


@dataclass(frozen=True)
class Preset:
    config: Path
    checkpoint: Path


SMALL = Preset(
    config=Path("src/config/AMT-S.yaml"),
    checkpoint=Path("src/pretrained/amt-s.pth"),
)

LARGE = Preset(
    config=Path("src/config/AMT-L.yaml"),
    checkpoint=Path("src/pretrained/amt-l.pth"),
)

GLOBAL = Preset(
    config=Path("src/config/AMT-g.yaml"),
    checkpoint=Path("src/pretrained/amt-g.pth"),
)