fix unrelated test

wangxuebing0906 · Jul 8, 2023 · dd4f48c · dd4f48c
1 parent aa95d22
commit dd4f48c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/data/memmap_dataset_test.py b/tests/data/memmap_dataset_test.py
@@ -16,7 +16,7 @@ def test_mmap_dataset(tmp_path: Path):
     mmap2[:] = np.array(list(range(16, 32)), dtype=np.uint16)
     mmap2.flush()
 
-    ds = MemMapDataset("/tmp/mmap1.npy", "/tmp/mmap2.npy", chunk_size=4)
+    ds = MemMapDataset(tmp_path / "mmap1.npy", tmp_path / "mmap2.npy", chunk_size=4)
     assert ds[0]["input_ids"].tolist() == [0, 1, 2, 3]
     assert ds[1]["input_ids"].tolist() == [4, 5, 6, 7]
     assert ds[7]["input_ids"].tolist() == [28, 29, 30, 31]