Skip to content

Commit b43e55c

Browse files
Fix Ruff rule B023
1 parent f011508 commit b43e55c

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

Diff for: tfx/components/example_diff/executor.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -175,7 +175,11 @@ def Do(self, input_dict: Dict[str, List[types.Artifact]],
175175
logging.info('Processing split pair %s', split_pair)
176176
# pylint: disable=cell-var-from-loop
177177
@beam.ptransform_fn
178-
def _iteration(p):
178+
def _iteration(p,
179+
test_tfxio=test_tfxio,
180+
base_tfxio=base_tfxio,
181+
split_pair=split_pair
182+
):
179183
base_examples = (
180184
p | 'TFXIORead[base]' >> test_tfxio.RawRecordBeamSource()
181185
| 'Parse[base]' >> beam.Map(_parse_example))

Diff for: tfx/dsl/placeholder/proto_placeholder.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -509,11 +509,11 @@ def _purge_types(
509509
_purge_types(name_prefix, message_descriptor) # Step 2
510510
_remove_unless(
511511
file_descriptor.message_type,
512-
lambda m: f'{name_prefix}.{m.name}' in self._keep_types, # pylint: disable=cell-var-from-loop
512+
lambda m, name_prefix=name_prefix: f'{name_prefix}.{m.name}' in self._keep_types, # pylint: disable=cell-var-from-loop
513513
)
514514
_remove_unless(
515515
file_descriptor.enum_type,
516-
lambda e: f'{name_prefix}.{e.name}' in self._keep_types, # pylint: disable=cell-var-from-loop
516+
lambda e, name_prefix=name_prefix: f'{name_prefix}.{e.name}' in self._keep_types, # pylint: disable=cell-var-from-loop
517517
)
518518

519519
# Step 4: Remove file descriptors that became empty. Remove declared

0 commit comments

Comments
 (0)