fix unit test

onnx · Jun 30, 2023 · 4c4b978 · 4c4b978
1 parent d253f70
commit 4c4b978
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_backend.py b/tests/test_backend.py
@@ -5038,7 +5038,7 @@ def func(x):
     def test_unique_with_counts_out_int32(self):
         x_val = np.array([2, 3, 3, 6, 4, 1, 1], dtype=np.float32)
         def func(x):
-            x1_, x2_, x3_ = tf.unique(x, out_idx=tf.int32)
+            x1_, x2_, x3_ = tf.unique_with_counts(x, out_idx=tf.int32)
             y1 = tf.identity(x1_, name=_TFOUTPUT)
             y2 = tf.identity(x2_, name=_TFOUTPUT1)
             y3 = tf.identity(x3_, name=_TFOUTPUT2)