Fix node

Signed-off-by: Justin Chu <justinchu@microsoft.com>
justinchuby · Sep 21, 2023 · f5bbf06 · f5bbf06
1 parent d2a9069
commit f5bbf06
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/onnx/backend/test/case/node/dft.py b/onnx/backend/test/case/node/dft.py
@@ -68,7 +68,7 @@ def export() -> None:
 
         x = x.reshape(1, 10, 10, 1)
         y = np.stack((y.real, y.imag), axis=2).astype(np.float32).reshape(1, 10, 10, 2)
-        expect(node, inputs=[x, "", axis], outputs=[y], name="test_dft")
+        expect(node, inputs=[x, axis], outputs=[y], name="test_dft")
 
         node = onnx.helper.make_node("DFT", inputs=["x"], outputs=["y"])
         x = np.arange(0, 100).reshape(10, 10).astype(np.float32)
@@ -77,7 +77,7 @@ def export() -> None:
 
         x = x.reshape(1, 10, 10, 1)
         y = np.stack((y.real, y.imag), axis=2).astype(np.float32).reshape(1, 10, 10, 2)
-        expect(node, inputs=[x, "", axis], outputs=[y], name="test_dft_axis")
+        expect(node, inputs=[x, axis], outputs=[y], name="test_dft_axis")
 
         node = onnx.helper.make_node("DFT", inputs=["x"], outputs=["y"], inverse=1)
         x = np.arange(0, 100, dtype=np.complex64).reshape(10, 10)
@@ -86,4 +86,4 @@ def export() -> None:
 
         x = np.stack((x.real, x.imag), axis=2).astype(np.float32).reshape(1, 10, 10, 2)
         y = np.stack((y.real, y.imag), axis=2).astype(np.float32).reshape(1, 10, 10, 2)
-        expect(node, inputs=[x, "", axis], outputs=[y], name="test_dft_inverse")
+        expect(node, inputs=[x, axis], outputs=[y], name="test_dft_inverse")