From a76682d74d742b5bafe83c8806f566d1c325cb73 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Carlos=20Mochol=C3=AD?= Date: Thu, 4 Apr 2024 18:36:59 +0200 Subject: [PATCH] Fix decorators --- tests/test_evaluate.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/test_evaluate.py b/tests/test_evaluate.py index 1418d168ad..408e5deca4 100644 --- a/tests/test_evaluate.py +++ b/tests/test_evaluate.py @@ -1,6 +1,5 @@ # Copyright Lightning AI. Licensed under the Apache License 2.0, see LICENSE file. -import os import shutil import subprocess import sys @@ -20,7 +19,11 @@ from litgpt.scripts.download import download_from_hub -@mock.patch.dict(os.environ, {"LT_ACCELERATOR": "cpu"}) +@pytest.mark.xfail( + raises=(datasets.builder.DatasetGenerationError, NotImplementedError), + strict=False, + match="Loading a dataset cached in a LocalFileSystem is not supported", +) def test_evaluate_script(tmp_path, monkeypatch): ours_config = Config.from_name("pythia-14m") download_from_hub(repo_id="EleutherAI/pythia-14m", tokenizer_only=True, checkpoint_dir=tmp_path)