From 87248027d04f0cd3d42d5c339f0af6a87f343e06 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sat, 14 Sep 2024 13:22:09 -0400 Subject: [PATCH] use revision tied to head --- tests/test_datasets.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_datasets.py b/tests/test_datasets.py index 5a631b2e6..845d7345b 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -279,7 +279,7 @@ class TestDatasetPreparation(unittest.TestCase): { "path": "mhenrichsen/alpaca_2k_test", "type": "alpaca", - "revision": "foo", + "revision": "d05c1cb", }, ], } @@ -319,7 +319,7 @@ class TestDatasetPreparation(unittest.TestCase): "data_files": [ "mhenrichsen/alpaca_2k_test/alpaca_2000.parquet", ], - "revision": "foo", + "revision": "d05c1cb", }, ], } @@ -335,5 +335,6 @@ class TestDatasetPreparation(unittest.TestCase): assert "labels" in dataset.features shutil.rmtree(tmp_ds_path) + if __name__ == "__main__": unittest.main()