[sharktank] Mark test expected to fail

Marks `testExportWithArgumentDeviceAffinities` as expected to fail if
running with torch>=2.4.0, see nod-ai#685.

sharktank/tests/export_test.py

@@ -4,6 +4,8 @@
 # See https://llvm.org/LICENSE.txt for license information.
 # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
+import pytest
+
 from sharktank.types import (
     ReplicatedTensor,
     SplitPrimitiveTensor,
@@ -70,6 +72,7 @@ def testGetFlatArgumentDeviceAffinities(self):
         }
         assert_dicts_equal(affinities, expected_affinities)
 
+    @pytest.mark.xfail(torch.__version__ >= (2,4), reason="https://github.com/nod-ai/shark-ai/issues/685")
     def testExportWithArgumentDeviceAffinities(self):
         args = (ReplicatedTensor(ts=[torch.tensor([1])]), torch.tensor([[2]]))