Revert "Low-Code CDK: make RecordFilter.filter_records as generator (#…

…24772)" This reverts commit 032f9b8.
airbytehq · Apr 19, 2023 · b571e4a · b571e4a
1 parent 2cdde75
commit b571e4a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/airbyte-cdk/python/airbyte_cdk/sources/declarative/extractors/record_filter.py b/airbyte-cdk/python/airbyte_cdk/sources/declarative/extractors/record_filter.py
@@ -33,6 +33,4 @@ def filter_records(
         next_page_token: Optional[Mapping[str, Any]] = None,
     ) -> List[Record]:
         kwargs = {"stream_state": stream_state, "stream_slice": stream_slice, "next_page_token": next_page_token}
-        for record in records:
-            if self._filter_interpolator.eval(self.config, record=record, **kwargs):
-                yield record
+        return [record for record in records if self._filter_interpolator.eval(self.config, record=record, **kwargs)]
diff --git a/airbyte-cdk/python/unit_tests/sources/declarative/extractors/test_record_filter.py b/airbyte-cdk/python/unit_tests/sources/declarative/extractors/test_record_filter.py
@@ -52,4 +52,4 @@ def test_record_filter(test_name, filter_template, records, expected_records):
     actual_records = record_filter.filter_records(
         records, stream_state=stream_state, stream_slice=stream_slice, next_page_token=next_page_token
     )
-    assert list(actual_records) == expected_records
+    assert actual_records == expected_records
diff --git a/airbyte-cdk/python/unit_tests/sources/declarative/extractors/test_record_selector.py b/airbyte-cdk/python/unit_tests/sources/declarative/extractors/test_record_selector.py
@@ -78,7 +78,7 @@ def test_record_filter(test_name, field_path, filter_template, body, expected_re
     actual_records = record_selector.select_records(
         response=response, stream_state=stream_state, stream_slice=stream_slice, next_page_token=next_page_token
     )
-    assert list(actual_records) == expected_records
+    assert actual_records == expected_records
 
 
 def create_response(body):