fix unit test

Signed-off-by: Salvetti, Francesco <[email protected]>

Author: f-salvetti

tests/test_loops.py

@@ -308,7 +308,7 @@ def test_while_loop_cond_subgraphs(self):
         def func(x):
             x_dim = tf.shape(x)[0]
             r = tf.cast(tf.zeros(1), x.dtype)
-            for i in range(tf.constant(10)):
+            for i in tf.range(10):
                 if i == x_dim:
                     break
                 r += x[i]
@@ -319,4 +319,4 @@ def func(x):
         self.run_test_case(func, feed_dict, input_names_with_port, output_names_with_port)
 
 if __name__ == '__main__':
-    unittest_main()
+    unittest_main()