Skip to content

Commit 477b5fc

Browse files
committed
Fix lints
1 parent 42e36fe commit 477b5fc

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
@@ -75,6 +75,8 @@
7575
from apache_beam.transforms.util import GcpHsmGeneratedSecret
7676
from apache_beam.transforms.util import GcpSecret
7777
from apache_beam.transforms.util import Secret
78+
from apache_beam.transforms.util import _BatchSizeEstimator
79+
from apache_beam.transforms.util import _GlobalWindowsBatchingDoFn
7880
from apache_beam.transforms.window import FixedWindows
7981
from apache_beam.transforms.window import GlobalWindow
8082
from apache_beam.transforms.window import GlobalWindows
@@ -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)