Skip to content

Commit f9ae89f

Browse files
authored
Skip a failing Bagua test for manual optimization (#16225)
1 parent 859a228 commit f9ae89f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/tests_pytorch/strategies/test_bagua_strategy.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ def test_bagua_default(tmpdir):
4545
assert isinstance(trainer.strategy, BaguaStrategy)
4646

4747

48+
@pytest.mark.xfail(raises=AssertionError, reason="Internal error in Bagua") # Unexpected rsp=<Response [500]'
4849
@RunIf(min_cuda_gpus=1, bagua=True)
4950
def test_manual_optimization(tmpdir):
5051
model = ManualOptimBoringModel()

0 commit comments

Comments
 (0)