fix cpu test

mosaicml · Apr 12, 2024 · ea56fa9 · ea56fa9
1 parent b9a5abc
commit ea56fa9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/a_scripts/train/test_train.py b/tests/a_scripts/train/test_train.py
@@ -147,6 +147,7 @@ def test_train_multi_eval(tmp_path: pathlib.Path):
         tuple)
 
 
+@pytest.mark.gpu
 def test_validate_config():
     conf_path: str = os.path.join(
         REPO_DIR,