Address comments

guestrin-lab · Nov 23, 2024 · 441d1ed · 441d1ed
2 parents 85d096c + 15f469c
commit 441d1ed
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/op_examples/extract.py b/examples/op_examples/extract.py
@@ -23,7 +23,7 @@
     "masked_col_2": "The age of the person",
 }
 
-new_df = df.sem_extract(input_cols, output_cols)
+new_df = df.sem_extract(input_cols, output_cols, extract_quotes=True)
 print(new_df)
 
 # A description can also be omitted for each output column

diff --git 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: dict[str, str | None],
-        extract_quotes: bool = True,
+        extract_quotes: bool = False,
         postprocessor: Callable[[list[str]], SemanticExtractPostprocessOutput] = extract_postprocess,
         return_raw_outputs: bool = False,
     ) -> pd.DataFrame: