Skip to content

Commit f67c3f2

Browse files
Remove module-level imports of extra deps in experimental.judges (#4598)
1 parent cb5fdf9 commit f67c3f2

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

trl/experimental/judges/judges.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -24,13 +24,6 @@
2424
from ...import_utils import is_llm_blender_available
2525

2626

27-
if is_llm_blender_available():
28-
import llm_blender
29-
30-
if is_openai_available():
31-
from openai import OpenAI
32-
33-
3427
DEFAULT_PAIRWISE_SYSTEM_PROMPT = '''I require a leaderboard for various large language models. I'll provide you with prompts given to these models and their corresponding outputs. Your task is to assess these responses, and select the model that produces the best output from a human perspective.
3528
3629
## Instruction
@@ -213,6 +206,8 @@ class PairRMJudge(BasePairwiseJudge):
213206
def __init__(self):
214207
if not is_llm_blender_available():
215208
raise ValueError("llm-blender is not installed. Please install it with `pip install llm-blender`.")
209+
import llm_blender
210+
216211
self.blender = llm_blender.Blender()
217212
self.blender.loadranker("llm-blender/PairRM", device=Accelerator().device)
218213

@@ -362,6 +357,8 @@ def __init__(
362357
):
363358
if not is_openai_available():
364359
raise ValueError("OpenAI client is not installed. Please install it with 'pip install openai'.")
360+
from openai import OpenAI
361+
365362
self.client = OpenAI()
366363
self.model = model
367364
self.system_prompt = system_prompt or DEFAULT_PAIRWISE_SYSTEM_PROMPT

0 commit comments

Comments
 (0)