Nit: list to sequence (#3678)

This commit is contained in:
Davis Chase
2023-04-27 14:41:59 -07:00
committed by GitHub
parent b807a114e4
commit 212aadd4af
+2 -2
View File
@@ -55,7 +55,7 @@ class StructuredQueryOutputParser(BaseOutputParser[StructuredQuery]):
return cls(ast_parse=ast_parser.parse)
def _format_attribute_info(info: List[AttributeInfo]) -> str:
def _format_attribute_info(info: Sequence[AttributeInfo]) -> str:
info_dicts = {}
for i in info:
i_dict = dict(i)
@@ -65,7 +65,7 @@ def _format_attribute_info(info: List[AttributeInfo]) -> str:
def _get_prompt(
document_contents: str,
attribute_info: List[AttributeInfo],
attribute_info: Sequence[AttributeInfo],
examples: Optional[List] = None,
allowed_comparators: Optional[Sequence[Comparator]] = None,
allowed_operators: Optional[Sequence[Operator]] = None,