Raise FutureWarning for classes moved to experimental

trl/trainer/bco_config.py

@@ -24,6 +24,8 @@ def __post_init__(self):
         warnings.warn(
             "The `BCOConfig` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.bco import BCOConfig`. The current import path will be removed and no longer "
-            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__post_init__()

trl/trainer/bco_trainer.py

@@ -24,6 +24,8 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `BCOTrainer` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.bco import BCOTrainer`. The current import path will be removed and no longer "
-            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)

trl/trainer/callbacks.py

@@ -271,7 +271,9 @@ def __init__(self, *args, **kwargs):
             "The `WinRateCallback` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.winrate_callback import WinRateCallback`. The current import path will be removed "
             "and no longer supported in TRL 0.29. For more information, see "
-            "https://github.com/huggingface/trl/issues/4223."
+            "https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)
 

trl/trainer/cpo_config.py

@@ -24,6 +24,8 @@ def __post_init__(self):
         warnings.warn(
             "The `CPOConfig` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.cpo import CPOConfig`. The current import path will be removed and no longer "
-            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__post_init__()

trl/trainer/gkd_config.py

@@ -24,6 +24,8 @@ def __post_init__(self):
         warnings.warn(
             "The `GKDConfig` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.gkd import GKDConfig`. The current import path will be removed and no longer "
-            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__post_init__()

trl/trainer/gkd_trainer.py

@@ -24,6 +24,8 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `GKDTrainer` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.gkd import GKDTrainer`. The current import path will be removed and no longer "
-            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)

trl/trainer/judges.py

@@ -29,7 +29,9 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `AllTrueJudge` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.judges import AllTrueJudge`. The current import path will be removed and no "
-            "longer supported in TRL 0.29."
+            "longer supported in TRL 0.29.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)
 
@@ -39,7 +41,9 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `BaseBinaryJudge` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.judges import BaseBinaryJudge`. The current import path will be removed and no "
-            "longer supported in TRL 0.29."
+            "longer supported in TRL 0.29.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)
 
@@ -49,7 +53,9 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `BaseJudge` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.judges import BaseJudge`. The current import path will be removed and no "
-            "longer supported in TRL 0.29."
+            "longer supported in TRL 0.29.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)
 
@@ -59,7 +65,9 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `BasePairwiseJudge` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.judges import BasePairwiseJudge`. The current import path will be removed and no "
-            "longer supported in TRL 0.29."
+            "longer supported in TRL 0.29.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)
 
@@ -69,7 +77,9 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `BaseRankJudge` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.judges import BaseRankJudge`. The current import path will be removed and no "
-            "longer supported in TRL 0.29."
+            "longer supported in TRL 0.29.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)
 
@@ -79,7 +89,9 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `HfPairwiseJudge` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.judges import HfPairwiseJudge`. The current import path will be removed and no "
-            "longer supported in TRL 0.29."
+            "longer supported in TRL 0.29.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)
 
@@ -89,7 +101,9 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `OpenAIPairwiseJudge` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.judges import OpenAIPairwiseJudge`. The current import path will be removed and no "
-            "longer supported in TRL 0.29."
+            "longer supported in TRL 0.29.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)
 
@@ -99,6 +113,8 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `PairRMJudge` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.judges import PairRMJudge`. The current import path will be removed and no "
-            "longer supported in TRL 0.29."
+            "longer supported in TRL 0.29.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)

trl/trainer/nash_md_config.py

@@ -24,6 +24,8 @@ def __post_init__(self):
         warnings.warn(
             "The `NashMDConfig` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.nash_md import NashMDConfig`. The current import path will be removed and no "
-            "longer supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "longer supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__post_init__()

trl/trainer/nash_md_trainer.py

@@ -24,6 +24,8 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `NashMDTrainer` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.nash_md import NashMDTrainer`. The current import path will be removed and no "
-            "longer supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "longer supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)

trl/trainer/online_dpo_config.py

@@ -25,6 +25,8 @@ def __post_init__(self):
             "The `OnlineDPOConfig` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.online_dpo import OnlineDPOConfig`. The current import path will be removed and "
             "no longer supported in TRL 0.29. For more information, see "
-            "https://github.com/huggingface/trl/issues/4223."
+            "https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__post_init__()

trl/trainer/online_dpo_trainer.py

@@ -25,6 +25,8 @@ def __init__(self, *args, **kwargs):
             "The `OnlineDPOTrainer` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.online_dpo import OnlineDPOTrainer`. The current import path will be removed and "
             "no longer supported in TRL 0.29. For more information, see "
-            "https://github.com/huggingface/trl/issues/4223."
+            "https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)

trl/trainer/orpo_config.py

@@ -24,6 +24,8 @@ def __post_init__(self):
         warnings.warn(
             "The `ORPOConfig` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.orpo import ORPOConfig`. The current import path will be removed and no longer "
-            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__post_init__()

trl/trainer/orpo_trainer.py

@@ -24,6 +24,8 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `ORPOTrainer` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.orpo import ORPOTrainer`. The current import path will be removed and no longer "
-            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)

trl/trainer/ppo_config.py

@@ -24,6 +24,8 @@ def __post_init__(self):
         warnings.warn(
             "The `PPOConfig` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.ppo import PPOConfig`. The current import path will be removed and no longer "
-            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__post_init__()

trl/trainer/ppo_trainer.py

@@ -24,6 +24,8 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `PPOTrainer` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.ppo import PPOTrainer`. The current import path will be removed and no longer "
-            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)

trl/trainer/prm_config.py

@@ -22,6 +22,8 @@ def __post_init__(self):
         warnings.warn(
             "The `PRMConfig` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.xco import PRMConfig`. The current import path will be removed and no longer "
-            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__post_init__()

trl/trainer/prm_trainer.py

@@ -22,6 +22,8 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `PRMTrainer` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.prm import PRMTrainer`. The current import path will be removed and no longer "
-            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)

trl/trainer/xpo_config.py

@@ -22,6 +22,8 @@ def __post_init__(self):
         warnings.warn(
             "The `XPOConfig` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.xco import XPOConfig`. The current import path will be removed and no longer "
-            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__post_init__()

trl/trainer/xpo_trainer.py

@@ -22,6 +22,8 @@ def __init__(self, *args, **kwargs):
         warnings.warn(
             "The `XPOTrainer` is now located in `trl.experimental`. Please update your imports to "
             "`from trl.experimental.xpo import XPOTrainer`. The current import path will be removed and no longer "
-            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223."
+            "supported in TRL 0.29. For more information, see https://github.com/huggingface/trl/issues/4223.",
+            FutureWarning,
+            stacklevel=2,
         )
         super().__init__(*args, **kwargs)