added test for TRAIN_SAVE_FILES frozen

understanding-search · Jul 26, 2024 · 4f6d781 · 4f6d781
1 parent ffa068b
commit 4f6d781
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/integration/test_training.py b/tests/integration/test_training.py
@@ -14,6 +14,10 @@
 from maze_transformer.training.wandb_logger import WandbJobType, WandbProject
 
 
+def test_train_save_files_frozen():
+    with pytest.raises(AttributeError):
+        TRAIN_SAVE_FILES.data_cfg = "new"
+
 @pytest.mark.usefixtures("temp_dir")
 def test_train_model_without_evals(temp_dir: Path):
     dataset = _create_dataset()