Update instruction_template_retriever.py

instruction_template_retriever.py

@@ -259,7 +259,7 @@ class InstructionTemplateRetriever:
         max_results=250,
         deduplicate=True,
         reweight=False,
-        reweighting_epsilon=True,
+        reweight_epsilon=0.05,
     ):
         """
         Given a document
@@ -277,7 +277,7 @@ class InstructionTemplateRetriever:
                 inp0, inp = itertools.tee(inp)
                 first_row = next(inp0)
                 r = Random(first_row[1].item())
-                epsilon = reweighting_epsilon
+                epsilon = reweight_epsilon
                 bucket = first_row[1]
                 items = []
                 weights = []