move tests to correct directory and cleanup slates examples

Author: olgavrou

libs/langchain/langchain/chains/rl_chain/__init__.py

@@ -1,9 +1,4 @@
 from .pick_best_chain import PickBest
-from .slates_chain import (
-    SlatesPersonalizerChain,
-    SlatesRandomPolicy,
-    SlatesFirstChoicePolicy,
-)
 from .rl_chain_base import (
     Embed,
     BasedOn,

libs/langchain/langchain/chains/rl_chain/requirements.txt

@@ -125,6 +125,7 @@ newspaper3k = {version = "^0.2.8", optional = true}
 amazon-textract-caller = {version = "<2", optional = true}
 xata = {version = "^1.0.0a7", optional = true}
 xmltodict = {version = "^0.13.0", optional = true}
+vowpal-wabbit-next = "0.6.0"
 
 
 [tool.poetry.group.test.dependencies]

libs/langchain/langchain/chains/rl_chain/tests/test_slates_text_embedder.py

@@ -1,8 +1,4 @@
-import sys
-
-sys.path.append("..")
-
-import rl_chain.pick_best_chain as pick_best_chain
+import langchain.chains.rl_chain.pick_best_chain as pick_best_chain
 from test_utils import MockEncoder
 import pytest
 from langchain.prompts.prompt import PromptTemplate

libs/langchain/langchain/chains/rl_chain/tests/test_utils.py

@@ -1,8 +1,4 @@
-import sys
-
-sys.path.append("..")
-
-import rl_chain.pick_best_chain as pick_best_chain
+import langchain.chains.rl_chain.pick_best_chain as pick_best_chain
 from test_utils import MockEncoder
 
 import pytest

libs/langchain/pyproject.toml

@@ -1,8 +1,4 @@
-import sys
-
-sys.path.append("..")
-
-import rl_chain.rl_chain_base as base
+import langchain.chains.rl_chain.rl_chain_base as base
 from test_utils import MockEncoder
 
 import pytest

libs/langchain/langchain/chains/rl_chain/tests/test_pick_best_chain_call.py renamed to libs/langchain/tests/unit_tests/chains/rl_chain/test_pick_best_chain_call.py

@@ -0,0 +1,3 @@
+class MockEncoder:
+    def encode(self, to_encode):
+        return "[encoded]" + to_encode