Skip to content

Commit 8be7201

Browse files
committed
Fix formatting
1 parent 0ce6ad5 commit 8be7201

2 files changed

Lines changed: 5 additions & 3 deletions

File tree

sdks/python/apache_beam/examples/inference/table_row_inference_batch.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -234,8 +234,7 @@ def run_batch_inference(
234234
logging.info(' Input: %s', input_file)
235235
logging.info(' Model: %s', model_path)
236236
logging.info(' Features: %s', feature_columns)
237-
logging.info(
238-
' Output: %s', output_table if output_table else output_file)
237+
logging.info(' Output: %s', output_table if output_table else output_file)
239238

240239
with beam.Pipeline(options=pipeline_options) as pipeline:
241240

sdks/python/apache_beam/examples/inference/table_row_inference_test.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,10 @@
3838

3939
# Module-level matcher for assert_that (must be picklable; no closure over self).
4040
REQUIRED_OUTPUT_KEYS = (
41-
'row_key', 'prediction', 'input_feature1', 'input_feature2',
41+
'row_key',
42+
'prediction',
43+
'input_feature1',
44+
'input_feature2',
4245
'input_feature3')
4346

4447

0 commit comments

Comments
 (0)