Fix import in rl

ProbablyFaiz · Aug 28, 2024 · 90db298 · 90db298
1 parent 618a5ad
commit 90db298
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/rl/llm/engines/client.py b/rl/llm/engines/client.py
@@ -6,12 +6,13 @@
 from tqdm.contrib.concurrent import thread_map
 
 import rl.utils.io
-from rl.llm.config import EngineFeature, LLMConfig
 from rl.llm.engines.core import (
     ChatInput,
+    EngineFeature,
     InferenceEngine,
     InferenceInput,
     InferenceOutput,
+    LLMConfig,
     apply_chat_template,
     register_engine,
 )

diff --git a/rl/llm/engines/local.py b/rl/llm/engines/local.py
@@ -15,11 +15,12 @@
 import tqdm
 
 import rl.utils.io
-from rl.llm.config import EngineFeature, LLMConfig
 from rl.llm.engines.core import (
+    EngineFeature,
     InferenceEngine,
     InferenceInput,
     InferenceOutput,
+    LLMConfig,
     apply_chat_template,
     register_engine,
 )