diff --git a/sharktank/tests/examples/main_test.py b/sharktank/tests/examples/main_test.py index fb43977df..18276a603 100644 --- 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.mark.skipif( + sys.platform == "win32", + reason="torch._dynamo.exc.Unsupported:" +) class ShardingTests(MainRunnerTestBase): def testExportFfnNet(self): from sharktank.examples.sharding.export_ffn_net import main