File tree Expand file tree Collapse file tree 1 file changed +0
-7
lines changed
airbyte_cdk/sources/streams/concurrent Expand file tree Collapse file tree 1 file changed +0
-7
lines changed Original file line number Diff line number Diff line change 44
55import functools
66import logging
7- import os
87from abc import ABC , abstractmethod
98from typing import (
109 Any ,
@@ -238,18 +237,12 @@ def _extract_cursor_value(self, record: Record) -> Any:
238237 return self ._connector_state_converter .parse_value (self ._cursor_field .extract_value (record ))
239238
240239 def close_partition (self , partition : Partition ) -> None :
241- test_env = os .getenv ("PYTEST_CURRENT_TEST" )
242- if test_env and "test_concurrent_declarative_source.py" in test_env :
243- LOGGER .info (f"Closing partition { partition .to_slice ()} " )
244- LOGGER .info (f"\t state before is { self ._concurrent_state } " )
245240 slice_count_before = len (self ._concurrent_state .get ("slices" , []))
246241 self ._add_slice_to_state (partition )
247242 if slice_count_before < len (
248243 self ._concurrent_state ["slices" ]
249244 ): # only emit if at least one slice has been processed
250245 self ._merge_partitions ()
251- if test_env and "test_concurrent_declarative_source.py" in test_env :
252- LOGGER .info (f"\t state after merged partition is { self ._concurrent_state } " )
253246 self ._emit_state_message ()
254247 self ._has_closed_at_least_one_slice = True
255248
You can’t perform that action at this time.
0 commit comments