fix shape_inference_test

Signed-off-by: shubhambhokare1 <[email protected]>
shubhambhokare1 · Jan 31, 2025 · db9878e · db9878e
1 parent 86b6a8d
commit db9878e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnx/test/shape_inference_test.py b/onnx/test/shape_inference_test.py
@@ -978,7 +978,7 @@ def test_resize_size(self, _, version) -> None:
             )
 
     @parameterized.expand(all_versions_for("RMSNormalization"))
-    def test_rms_normalization(self) -> None:
+    def test_rms_normalization(self _, version) -> None:
         graph = self._make_graph(
             [
                 ("X", TensorProto.FLOAT, ("N", "C", "H", "W")),