tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

use revision tied to head

Browse Source
This commit is contained in:
Wing Lian
2024-09-14 13:22:09 -04:00
parent d0d22b7812
commit 87248027d0
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_datasets.py
View File

@@ -279,7 +279,7 @@ class TestDatasetPreparation(unittest.TestCase):
{ {
"path": "mhenrichsen/alpaca_2k_test", "path": "mhenrichsen/alpaca_2k_test",
"type": "alpaca", "type": "alpaca",
"revision": "foo", "revision": "d05c1cb",
}, },
], ],
} }
@@ -319,7 +319,7 @@ class TestDatasetPreparation(unittest.TestCase):
"data_files": [ "data_files": [
"mhenrichsen/alpaca_2k_test/alpaca_2000.parquet", "mhenrichsen/alpaca_2k_test/alpaca_2000.parquet",
], ],
"revision": "foo", "revision": "d05c1cb",
}, },
], ],
} }
@@ -335,5 +335,6 @@ class TestDatasetPreparation(unittest.TestCase):
assert "labels" in dataset.features assert "labels" in dataset.features
shutil.rmtree(tmp_ds_path) shutil.rmtree(tmp_ds_path)
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main()