From a19f3b8e9df1a43284a4dfb3cc4e42e7a728102d Mon Sep 17 00:00:00 2001 From: Mia Dai Date: Sun, 23 Nov 2025 09:41:30 -0600 Subject: [PATCH 1/2] Change bool to int Signed-off-by: Mia Dai --- beast/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/beast/train.py b/beast/train.py index bfa386e..80204b7 100644 --- a/beast/train.py +++ b/beast/train.py @@ -165,7 +165,7 @@ def train(config: dict, model, output_dir: str | Path): def get_callbacks( checkpointing: bool = True, lr_monitor: bool = True, - ckpt_every_n_epochs: bool = None, + ckpt_every_n_epochs: int = None, ) -> list: callbacks = [] From c82e88ca5f3dd204cd00f7c86cc3386ad37e0dd2 Mon Sep 17 00:00:00 2001 From: Mia Dai Date: Mon, 24 Nov 2025 08:06:38 -0600 Subject: [PATCH 2/2] add None to type --- beast/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/beast/train.py b/beast/train.py index 80204b7..8258d4a 100644 --- a/beast/train.py +++ b/beast/train.py @@ -165,7 +165,7 @@ def train(config: dict, model, output_dir: str | Path): def get_callbacks( checkpointing: bool = True, lr_monitor: bool = True, - ckpt_every_n_epochs: int = None, + ckpt_every_n_epochs: int | None= None, ) -> list: callbacks = []