update test too

axolotl-ai-cloud · Apr 21, 2024 · e122f4d · e122f4d
1 parent 369905f
commit e122f4d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_datasets.py b/tests/test_datasets.py
@@ -110,7 +110,7 @@ def test_load_from_save_to_disk(self):
         """Usual use case.  Verify datasets saved via `save_to_disk` can be loaded."""
         with tempfile.TemporaryDirectory() as tmp_dir:
             tmp_ds_name = Path(tmp_dir) / "tmp_dataset"
-            self.dataset.save_to_disk(tmp_ds_name)
+            self.dataset.save_to_disk(str(tmp_ds_name))
 
             prepared_path = Path(tmp_dir) / "prepared"
             cfg = DictDefault(