[train] Fix HuggingFace -> Transformers wrapping logic (ray-project#3…

…5276) Properly pass constructor arguments through. Signed-off-by: Matthew Deng <[email protected]> Signed-off-by: Marcus Zhang <[email protected]>
alipay · May 12, 2023 · a72c1ee · a72c1ee
1 parent b907ed9
commit a72c1ee
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/python/ray/train/huggingface/huggingface_checkpoint.py b/python/ray/train/huggingface/huggingface_checkpoint.py
@@ -14,7 +14,7 @@ class HuggingFaceCheckpoint(TransformersCheckpoint):
     # than __init__
     def __new__(cls: type, *args, **kwargs):
         warnings.warn(deprecation_msg, DeprecationWarning)
-        return super(HuggingFaceCheckpoint, cls).__new__(cls)
+        return super(HuggingFaceCheckpoint, cls).__new__(cls, *args, **kwargs)
 
 
 __all__ = [

diff --git a/python/ray/train/huggingface/huggingface_predictor.py b/python/ray/train/huggingface/huggingface_predictor.py
@@ -14,7 +14,7 @@ class HuggingFacePredictor(TransformersPredictor):
     # than __init__
     def __new__(cls: type, *args, **kwargs):
         warnings.warn(deprecation_msg, DeprecationWarning)
-        return super(HuggingFacePredictor, cls).__new__(cls)
+        return super(HuggingFacePredictor, cls).__new__(cls, *args, **kwargs)
 
 
 __all__ = [

diff --git a/python/ray/train/huggingface/huggingface_trainer.py b/python/ray/train/huggingface/huggingface_trainer.py
@@ -14,7 +14,7 @@ class HuggingFaceTrainer(TransformersTrainer):
     # than __init__
     def __new__(cls: type, *args, **kwargs):
         warnings.warn(deprecation_msg, DeprecationWarning)
-        return super(HuggingFaceTrainer, cls).__new__(cls)
+        return super(HuggingFaceTrainer, cls).__new__(cls, *args, **kwargs)
 
 
 __all__ = [