From 15f469cb48e11d45b3ee28b7b8dc18dc06edd189 Mon Sep 17 00:00:00 2001 From: liana313 Date: Sat, 23 Nov 2024 09:38:18 -0800 Subject: [PATCH] update quoting in extract to non-default behavior --- examples/op_examples/extract.py | 2 +- lotus/sem_ops/sem_extract.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/op_examples/extract.py b/examples/op_examples/extract.py index c8b9f171..23dda951 100644 --- a/examples/op_examples/extract.py +++ b/examples/op_examples/extract.py @@ -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) diff --git a/lotus/sem_ops/sem_extract.py b/lotus/sem_ops/sem_extract.py index fb0df1c0..f1afcb9c 100644 --- a/lotus/sem_ops/sem_extract.py +++ b/lotus/sem_ops/sem_extract.py @@ -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: