From 4e23d958f20fd4fdd795cb06c2cdb7ebea704855 Mon Sep 17 00:00:00 2001
From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>
Date: Mon, 17 Jun 2024 16:41:36 +0200
Subject: [PATCH] fix red CI

---
 trl/trainer/cpo_config.py | 1 +
 trl/trainer/dpo_config.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/trl/trainer/cpo_config.py b/trl/trainer/cpo_config.py
index d231fa83cc..99c74e2c87 100644
--- a/trl/trainer/cpo_config.py
+++ b/trl/trainer/cpo_config.py
@@ -83,3 +83,4 @@ class CPOConfig(TrainingArguments):
     def __post_init__(self):
         if self.loss_type == "kto_pair":
             raise ValueError("Support for kto_pair has been removed in CPOTrainer. Please use KTOTrainer.")
+        return super().__post_init__()
\ No newline at end of file
diff --git a/trl/trainer/dpo_config.py b/trl/trainer/dpo_config.py
index 88b8b2d1fb..38857409b0 100644
--- a/trl/trainer/dpo_config.py
+++ b/trl/trainer/dpo_config.py
@@ -106,3 +106,4 @@ class DPOConfig(TrainingArguments):
     def __post_init__(self):
         if self.loss_type == "kto_pair":
             raise ValueError("Support for kto_pair has been removed in DPOTrainer. Please use KTOTrainer.")
+        return super().__post_init__()