From 5d791465f8408f391ef4708c4ed2e53d21f1b642 Mon Sep 17 00:00:00 2001 From: Ben Auffarth Date: Sat, 16 Aug 2025 17:07:54 +0100 Subject: [PATCH 1/2] requirements for training datasets --- requirements.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/requirements.txt b/requirements.txt index 8c90d6f6..e0a10367 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,3 +9,4 @@ wandb omegaconf hydra-core huggingface_hub +numpy From 43bd4bd59abae4b1d6f6f9923c1041fb79147f0a Mon Sep 17 00:00:00 2001 From: Ben Auffarth Date: Sat, 16 Aug 2025 17:08:16 +0100 Subject: [PATCH 2/2] clearer messages --- dataset/build_sudoku_dataset.py | 1 + 1 file changed, 1 insertion(+) diff --git a/dataset/build_sudoku_dataset.py b/dataset/build_sudoku_dataset.py index 7924438b..0f1a54b1 100644 --- a/dataset/build_sudoku_dataset.py +++ b/dataset/build_sudoku_dataset.py @@ -58,6 +58,7 @@ def apply_transformation(x: np.ndarray) -> np.ndarray: def convert_subset(set_name: str, config: DataProcessConfig): + print(f"Create {set_name} dataset") # Read CSV inputs = [] labels = []