Skip to content

Commit

Permalink
update quoting in extract to non-default behavior
Browse files Browse the repository at this point in the history
  • Loading branch information
liana313 committed Nov 23, 2024
1 parent 654290d commit 15f469c
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion examples/op_examples/extract.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,5 +18,5 @@
input_cols = ["description"]
output_cols = ["name", "age"]

new_df = df.sem_extract(input_cols, output_cols)
new_df = df.sem_extract(input_cols, output_cols, extract_quotes=True)
print(new_df)
2 changes: 1 addition & 1 deletion lotus/sem_ops/sem_extract.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ def __call__(
self,
input_cols: list[str],
output_cols: list[str],
extract_quotes: bool = True,
extract_quotes: bool = False,
postprocessor: Callable[[list[str]], SemanticExtractPostprocessOutput] = extract_postprocess,
return_raw_outputs: bool = False,
) -> pd.DataFrame:
Expand Down

0 comments on commit 15f469c

Please sign in to comment.