Update instruction_template_retriever.py

instruction_template_retriever.py

@@ -270,6 +270,8 @@ class InstructionTemplateRetriever:
             search_k (int): The number of search results to pull when retrieving from FAISS.
             max_results (int): The max number of results to return.
             deduplicate (bool): Deduplicate results between coverage sections.
+            reweight (bool): Whether to reweight the results based on a more realistic length distribution.
+            reweight_epsilon (float): How tolerant to be when reweighting (larger is more inaccurate results but better reweighting)
         """
 
         def _reweight(inp, k=None):