Marked test for [Arg]Sort Op in PyTorch as xfail

Author: twaclaw

tests/link/pytorch/test_sort.py

@@ -7,17 +7,12 @@
 from tests.link.pytorch.test_basic import compare_pytorch_and_py
 
 
+@pytest.mark.xfail(reason="Reshape not implemented")
 @pytest.mark.parametrize("axis", [0, 1, None])
 @pytest.mark.parametrize("func", (sort, argsort))
 def test_sort(func, axis):
     x = matrix("x", shape=(2, 2), dtype="float64")
     out = func(x, axis=axis)
     fgraph = FunctionGraph([x], [out])
     arr = np.array([[1.0, 4.0], [5.0, 2.0]])
-
-    # TODO: remove condition once Reshape is implemented
-    if axis is None:
-        with pytest.raises(NotImplementedError):
-            compare_pytorch_and_py(fgraph, [arr])
-    else:
-        compare_pytorch_and_py(fgraph, [arr])
+    compare_pytorch_and_py(fgraph, [arr])