diff --git a/chatnoir_pyterrier/retrieve.py b/chatnoir_pyterrier/retrieve.py index 6b808cd..5cdda56 100644 --- a/chatnoir_pyterrier/retrieve.py +++ b/chatnoir_pyterrier/retrieve.py @@ -137,6 +137,7 @@ def _transform_query(self, topic: DataFrame) -> DataFrame: explain=explain, staging=self.staging, page_size=page_size, + retries=self.retries, ).results else: results = search_phrases( @@ -147,6 +148,7 @@ def _transform_query(self, topic: DataFrame) -> DataFrame: explain=explain, staging=self.staging, page_size=page_size, + retries=self.retries, ).results if self.filter_unknown: