Apply suggestions from code review

Co-authored-by: matthewdeng <matthew.j.deng@gmail.com> Signed-off-by: Yunxuan Xiao <xiaoyunxuan1998@gmail.com>
ray-project · Aug 9, 2023 · 7957cf1 · 7957cf1
1 parent 6ed560b
commit 7957cf1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/ray/train/huggingface/transformers/_transformers_utils.py b/python/ray/train/huggingface/transformers/_transformers_utils.py
@@ -270,7 +270,7 @@ def prepare_trainer(trainer: Trainer) -> Trainer:
     """Prepare your HuggingFace Transformer Trainer for Ray Train."""
     base_trainer_class: Type[transformers.trainer.Trainer] = trainer.__class__
 
-    class RayTrainer(base_trainer_class):
+    class RayTransformersTrainer(base_trainer_class):
         """A Wrapper of `transformers.Trainer` for Ray Data Integration."""
 
         def get_train_dataloader(self):