diff --git a/experimental/torch_xla2/test/test_ops.py b/experimental/torch_xla2/test/test_ops.py index 825ea8a12af..d40189fd0a8 100644 --- a/experimental/torch_xla2/test/test_ops.py +++ b/experimental/torch_xla2/test/test_ops.py @@ -320,6 +320,7 @@ "cumulative_trapezoid", "expand_as", "mean", + "nanmean", "atleast_1d", "atleast_2d", "atleast_3d",