[data] add eval_on_each_dataset arg (#7912)

2025-04-30 06:56:43 +08:00
parent 6d2cde43e7
commit d4ee44bdef
3 changed files with 15 additions and 5 deletions
--- a/src/llamafactory/hparams/training_args.py
+++ b/src/llamafactory/hparams/training_args.py
@@ -64,6 +64,7 @@ class RayArguments:
                raise ValueError(
                    f"ray_storage_filesystem must be one of ['s3', 'gs', 'gcs'], got {self.ray_storage_filesystem}"
                )
+
            import pyarrow.fs as fs

            if self.ray_storage_filesystem == "s3":