[sharktank] Skip testExportFfnNet on Windows

nod-ai · Dec 16, 2024 · e10c30d · e10c30d
1 parent 0804746
commit e10c30d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/sharktank/tests/examples/main_test.py b/sharktank/tests/examples/main_test.py
@@ -4,11 +4,17 @@
 # See https://llvm.org/LICENSE.txt for license information.
 # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
+import pytest
+import sys
 import unittest
 
 from sharktank.utils.testing import MainRunnerTestBase
 
 
+@pytest.skipif(
+    sys.platform == "win32",
+    reason="torch._dynamo.exc.Unsupported:"
+)
 class ShardingTests(MainRunnerTestBase):
     def testExportFfnNet(self):
         from sharktank.examples.sharding.export_ffn_net import main