From 67dc1a9dd20ab4a6c41498962fa3bd6e5ae1e710 Mon Sep 17 00:00:00 2001
From: olgavrou <olgavrou@gmail.com>
Date: Mon, 4 Sep 2023 07:36:47 -0400
Subject: [PATCH] cleanup

---
 libs/langchain/langchain/chains/rl_chain/__init__.py        | 2 ++
 libs/langchain/langchain/chains/rl_chain/pick_best_chain.py | 2 --
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/libs/langchain/langchain/chains/rl_chain/__init__.py b/libs/langchain/langchain/chains/rl_chain/__init__.py
index 3a14861bd7..80242139f5 100644
--- a/libs/langchain/langchain/chains/rl_chain/__init__.py
+++ b/libs/langchain/langchain/chains/rl_chain/__init__.py
@@ -15,6 +15,7 @@ from langchain.chains.rl_chain.base import (
 from langchain.chains.rl_chain.pick_best_chain import (
     PickBest,
     PickBestEvent,
+    PickBestFeatureEmbedder,
     PickBestSelected,
 )
 
@@ -37,6 +38,7 @@ __all__ = [
     "PickBest",
     "PickBestEvent",
     "PickBestSelected",
+    "PickBestFeatureEmbedder",
     "Embed",
     "BasedOn",
     "ToSelectFrom",
diff --git a/libs/langchain/langchain/chains/rl_chain/pick_best_chain.py b/libs/langchain/langchain/chains/rl_chain/pick_best_chain.py
index e3e93b138e..afc7fc1e30 100644
--- a/libs/langchain/langchain/chains/rl_chain/pick_best_chain.py
+++ b/libs/langchain/langchain/chains/rl_chain/pick_best_chain.py
@@ -60,9 +60,7 @@ class PickBestFeatureEmbedder(base.Embedder[PickBestEvent]):
 
         if model is None:
             from sentence_transformers import SentenceTransformer
-
             model = SentenceTransformer("all-mpnet-base-v2")
-            # model = SentenceTransformer("all-MiniLM-L6-v2")
 
         self.model = model
         self.auto_embed = auto_embed