File tree Expand file tree Collapse file tree
sdks/python/apache_beam/transforms Expand file tree Collapse file tree Original file line number Diff line number Diff line change 7272from apache_beam .transforms .core import FlatMapTuple
7373from apache_beam .transforms .trigger import AfterCount
7474from apache_beam .transforms .trigger import Repeatedly
75+ from apache_beam .transforms .util import _BatchSizeEstimator
76+ from apache_beam .transforms .util import _GlobalWindowsBatchingDoFn
7577from apache_beam .transforms .util import GcpHsmGeneratedSecret
7678from apache_beam .transforms .util import GcpSecret
7779from apache_beam .transforms .util import Secret
@@ -1261,8 +1263,7 @@ def check_batch_homogeneity(batch):
12611263 def test_global_batching_dofn_single_vs_multiple_bundles (self ):
12621264 # This test directly verifies how bundling affects the batch sizes produced by
12631265 # the internal _GlobalWindowsBatchingDoFn of BatchElements.
1264- from apache_beam .transforms .util import _GlobalWindowsBatchingDoFn
1265- from apache_beam .transforms .util import _BatchSizeEstimator
1266+
12661267
12671268 # 1. Single Bundle Scenario:
12681269 # Four elements processed within the same start_bundle / finish_bundle lifecycle.
You can’t perform that action at this time.
0 commit comments