style, fix an issue with tensorflow 2.1

Signed-off-by: xavier dupré <[email protected]>

Author: sdpython

examples/end2end_tfkeras.py

@@ -5,6 +5,7 @@
 discrepencies. Inferencing time is also compared between
 *onnxruntime*, *tensorflow* and *tensorflow.lite*.
 """
+from onnxruntime import InferenceSession
 import os
 import subprocess
 import timeit
@@ -37,7 +38,7 @@
 ########################################
 # Testing the model.
 input = np.random.randn(2, 4, 4).astype(np.float32)
-expected  = model.predict(input)
+expected = model.predict(input)
 print(expected)
 
 ########################################
@@ -57,7 +58,6 @@
 
 ########################################
 # Runs onnxruntime.
-from onnxruntime import InferenceSession
 session = InferenceSession("simple_rnn.onnx")
 got = session.run(None, {'input_1:0': input})
 print(got[0])
@@ -79,7 +79,7 @@
 tflite_model = converter.convert()
 with open("simple_rnn.tflite", "wb") as f:
     f.write(tflite_model)
-    
+
 # Builds an interpreter
 interpreter = tf.lite.Interpreter(model_path='simple_rnn.tflite')
 interpreter.allocate_tensors()
@@ -89,14 +89,16 @@
 print("output_details", output_details)
 index = input_details[0]['index']
 
+
 def tflite_predict(input, interpreter=interpreter, index=index):
     res = []
     for i in range(input.shape[0]):
-        interpreter.set_tensor(index, input[i:i+1])
+        interpreter.set_tensor(index, input[i:i + 1])
         interpreter.invoke()
         res.append(interpreter.get_tensor(output_details[0]['index']))
     return np.vstack(res)
 
+
 print(input[0:1].shape, "----", input_details[0]['shape'])
 output_data = tflite_predict(input, interpreter, index)
 print(output_data)

tests/test_example.py

@@ -29,7 +29,7 @@ def run_example(self, name, expected=None):
         err = proc.stderr.decode('ascii')
         self.assertTrue(err is not None)
 
-    @check_tf_min_version("2.0", "use tf.keras")
+    @check_tf_min_version("2.3", "use tf.keras")
     @check_opset_min_version(12)
     @check_opset_max_version(13)
     def test_end2end_tfkeras(self):