Skip to content

Commit 645ae6d

Browse files
committed
Fix lints
1 parent 42e36fe commit 645ae6d

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

sdks/python/apache_beam/transforms/util_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,8 @@
7272
from apache_beam.transforms.core import FlatMapTuple
7373
from apache_beam.transforms.trigger import AfterCount
7474
from apache_beam.transforms.trigger import Repeatedly
75+
from apache_beam.transforms.util import _BatchSizeEstimator
76+
from apache_beam.transforms.util import _GlobalWindowsBatchingDoFn
7577
from apache_beam.transforms.util import GcpHsmGeneratedSecret
7678
from apache_beam.transforms.util import GcpSecret
7779
from apache_beam.transforms.util import Secret
@@ -1261,8 +1263,6 @@ 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
12661266

12671267
# 1. Single Bundle Scenario:
12681268
# Four elements processed within the same start_bundle / finish_bundle lifecycle.

0 commit comments

Comments
 (0)