diff --git a/src/gabriel/tasks/discover.py b/src/gabriel/tasks/discover.py index ee1fa3a..34a6dcf 100644 --- a/src/gabriel/tasks/discover.py +++ b/src/gabriel/tasks/discover.py @@ -211,6 +211,7 @@ async def run( categories=None, additional_instructions=self.cfg.additional_instructions or "", reset_files=reset_files, + **kwargs, ) term_defs: Dict[str, str] = {} if "coded_passages" in codify_df: @@ -249,6 +250,7 @@ async def run( circle_column_name, # type: ignore[arg-type] square_column_name, # type: ignore[arg-type] reset_files=reset_files, + **kwargs, ) term_defs = {} for attr, expl in zip( @@ -293,6 +295,7 @@ async def run( candidate_df, "term", reset_files=reset_files, + **kwargs, ) labels = ( @@ -390,6 +393,7 @@ def derive_base_from_combined( circle_column_name=circle_column_name, # type: ignore[arg-type] square_column_name=square_column_name, # type: ignore[arg-type] reset_files=reset_files, + **kwargs, ) actual_combined_labels = dict(clf.cfg.labels) @@ -489,6 +493,7 @@ def derive_base_from_combined( df, column_name, # type: ignore[arg-type] reset_files=reset_files, + **kwargs, ) actual_labels = dict(clf.cfg.labels) if actual_labels != labels: