Skip to content

Commit a1817a1

Browse files
committed
Remove trailing whitespace
Signed-off-by: Guoqing Bao <[email protected]>
1 parent b62cdf4 commit a1817a1

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

tf2onnx/convert.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -476,12 +476,12 @@ def wrap_call(*args, training=False, **kwargs):
476476
# captured_inputs = [t_name.name for t_val, t_name in graph_captures.values()]
477477

478478
if hasattr(concrete_func.graph, "captures"): #for Tensorflow 2.13+
479-
graph_captures = concrete_func.graph.captures
479+
graph_captures = concrete_func.graph.captures
480480
captured_inputs = [t_name.name for t_val, t_name in graph_captures]
481481
else:
482-
graph_captures = concrete_func.graph._captures # pylint: disable=protected-access
482+
graph_captures = concrete_func.graph._captures # pylint: disable=protected-access
483483
captured_inputs = [t_name.name for t_val, t_name in graph_captures.values()]
484-
484+
485485
input_names = [input_tensor.name for input_tensor in concrete_func.inputs
486486
if input_tensor.name not in captured_inputs]
487487
output_names = [output_tensor.name for output_tensor in concrete_func.outputs

tf2onnx/tf_loader.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -571,10 +571,10 @@ def _from_saved_model_v2(model_path, input_names, output_names, tag, signature_d
571571
if input_names is None:
572572
inputs = [tensor.name for tensor in concrete_func.inputs if tensor.dtype != tf.dtypes.resource]
573573
if hasattr(concrete_func.graph, "captures"): #for Tensorflow 2.13+
574-
graph_captures = concrete_func.graph.captures
574+
graph_captures = concrete_func.graph.captures
575575
captured_inputs = [t_name.name for t_val, t_name in graph_captures]
576576
else:
577-
graph_captures = concrete_func.graph._captures # pylint: disable=protected-access
577+
graph_captures = concrete_func.graph._captures # pylint: disable=protected-access
578578
captured_inputs = [t_name.name for t_val, t_name in graph_captures.values()]
579579

580580
# graph_captures = concrete_func.graph._captures # pylint: disable=protected-access

0 commit comments

Comments
 (0)