Skip to content

Commit

Permalink
Fix decorators
Browse files Browse the repository at this point in the history
  • Loading branch information
carmocca committed Apr 4, 2024
1 parent 8dd203f commit a76682d
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions tests/test_evaluate.py
Original file line number Diff line number Diff line change
@@ -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
Expand All @@ -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)
Expand Down

0 comments on commit a76682d

Please sign in to comment.