[console_scripts]
ketos = kraken.ketos:cli
kraken = kraken.kraken:cli

[ketos.cli]
compile = kraken.ketos.dataset:compile
convert = kraken.ketos.weights:convert
pretrain = kraken.ketos.pretrain:pretrain
publish = kraken.ketos.repo:publish
roadd = kraken.ketos.ro:roadd
rotrain = kraken.ketos.ro:rotrain
segtest = kraken.ketos.segmentation:segtest
segtrain = kraken.ketos.segmentation:segtrain
test = kraken.ketos.recognition:test
train = kraken.ketos.recognition:train

[kraken.cli]
binarize = kraken.kraken:binarize
get = kraken.kraken:get
list = kraken.kraken:list_models
ocr = kraken.kraken:ocr
segment = kraken.kraken:segment
show = kraken.kraken:show

[kraken.configs]
BLLASegmentationTrainingConfig = kraken.configs.vgsl:BLLASegmentationTrainingConfig
BLLASegmentationTrainingDataConfig = kraken.configs.vgsl:BLLASegmentationTrainingDataConfig
Config = kraken.configs.base:Config
ROTrainingConfig = kraken.configs.ro:ROTrainingConfig
ROTrainingDataConfig = kraken.configs.ro:ROTrainingDataConfig
RecognitionTrainingDataConfig = kraken.configs.base:RecognitionTrainingDataConfig
SegmentationTrainingDataConfig = kraken.configs.base:SegmentationTrainingDataConfig
TrainingConfig = kraken.configs.base:TrainingConfig
TrainingDataConfig = kraken.configs.base:TrainingDataConfig
VGSLPreTrainingConfig = kraken.configs.pretrain:VGSLPreTrainingConfig
VGSLPreTrainingDataConfig = kraken.configs.pretrain:VGSLPreTrainingDataConfig
VGSLRecognitionTrainingConfig = kraken.configs.vgsl:VGSLRecognitionTrainingConfig
VGSLRecognitionTrainingDataConfig = kraken.configs.vgsl:VGSLRecognitionTrainingDataConfig

[kraken.lightning_modules]
blla = kraken.train:BLLASegmentationModel
pretrain = kraken.lib.pretrain:RecognitionPretrainModel
ro = kraken.lib.ro:ROModel
vgsl = kraken.train:VGSLRecognitionModel

[kraken.loaders]
coreml = kraken.models:load_coreml
safetensors = kraken.models:load_safetensors

[kraken.models]
ROMLP = kraken.lib.ro:ROMLP
TorchVGSLModel = kraken.lib.vgsl:TorchVGSLModel
Wav2Vec2Mask = kraken.lib.pretrain:Wav2Vec2Mask

[kraken.tasks]
alignment = kraken.tasks:ForcedAlignmentTaskModel
recognition = kraken.tasks:RecognitionTaskModel
segmentation = kraken.tasks:SegmentationTaskModel

[kraken.writers]
coreml = kraken.models:write_coreml
safetensors = kraken.models:write_safetensors
