From 4418e030c2c4439dd4ae158887f8a6b64b6a1ff6 Mon Sep 17 00:00:00 2001 From: Wiktor Adamski Date: Fri, 15 Feb 2019 14:14:09 +0100 Subject: [PATCH] Changed code formatting. --- mace/python/tools/converter_tool/tensorflow_converter.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/mace/python/tools/converter_tool/tensorflow_converter.py b/mace/python/tools/converter_tool/tensorflow_converter.py index fa14eefd..d5be0ee8 100644 --- a/mace/python/tools/converter_tool/tensorflow_converter.py +++ b/mace/python/tools/converter_tool/tensorflow_converter.py @@ -205,8 +205,8 @@ class TensorflowConverter(base_converter.ConverterInterface): } pad_type = { - 'CONSTANT' : PadType.CONSTANT, - 'REFLECT' : PadType.REFLECT, + 'CONSTANT': PadType.CONSTANT, + 'REFLECT': PadType.REFLECT, 'SYMMETRIC': PadType.SYMMETRIC } @@ -740,7 +740,8 @@ class TensorflowConverter(base_converter.ConverterInterface): if len(tf_op.inputs) == 3: constant_value_arg = op.arg.add() constant_value_arg.name = MaceKeyword.mace_constant_value_str - constant_value = tf_op.inputs[2].eval().astype(np.int32).flat[0] + constant_value = tf_op.inputs[2].eval().astype(np.int32) \ + .flat[0] constant_value_arg.i = constant_value self._skip_tensor.add(tf_op.inputs[2].name) -- GitLab