Skip to content

Commit

Permalink
fix gpu CI test failure (#401)
Browse files Browse the repository at this point in the history
Signed-off-by: Shuoyang Ding <[email protected]>
  • Loading branch information
shuoyangd authored Dec 2, 2024
1 parent d14ac42 commit bc724ec
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions tests/test_read_simple_bitext.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,13 +21,15 @@

# The source/target file paths will be concatenated as document ID in `ParallelDataset`, so we can't directly pass a `Path` object.
@pytest.fixture
def src_file():
return Path("tests/bitext_data/toy.de").absolute().as_posix()
def src_file(pytestconfig):
root_dir = Path(pytestconfig.rootdir)
return (root_dir / "tests" / "bitext_data" / "toy.de").absolute().as_posix()


@pytest.fixture
def tgt_file():
return Path("tests/bitext_data/toy.en").absolute().as_posix()
def tgt_file(pytestconfig):
root_dir = Path(pytestconfig.rootdir)
return (root_dir / "tests" / "bitext_data" / "toy.en").absolute().as_posix()


class TestReadSimpleBitext:
Expand Down

0 comments on commit bc724ec

Please sign in to comment.