Update on "[ONNX] Support aten::scatter_reduce"

Fixes #84260 `reduce='mean'` is not supported, as it's not in ONNX spec (onnx/onnx#5100) [ghstack-poisoned]
pytorch · May 25, 2023 · c61b247 · c61b247
2 parents 313e606 + 00f5e9c
commit c61b247
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/onnx/test_pytorch_onnx_onnxruntime.py b/test/onnx/test_pytorch_onnx_onnxruntime.py
@@ -736,8 +736,8 @@ def forward(
 
         # Optional supports None inputs
         self.run_test(model, (x,))
-        # TODO: AssertionError: Tensor-likes are not close!
-        # This might be the first arg x is skipped.
+        # NOTE: default value is not supported on ONNX, so torch and ONNX has
+        # different behavior
         with self.assertRaisesRegex(AssertionError, "Tensor-likes are not close!"):
             self.run_test(model, (), {"y": y}, input_names=["y"])