Enable cp tests

Signed-off-by: Jerome <[email protected]>
Lightning-AI · Oct 15, 2024 · d1364d4 · d1364d4
1 parent 0d8247d
commit d1364d4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.azure/hpu-tests.yml b/.azure/hpu-tests.yml
@@ -117,7 +117,7 @@ jobs:
           tests/test_pytorch/test_precision.py \
           tests/test_pytorch/strategies/test_hpu_parallel.py \
           tests/test_pytorch/strategies/test_hpu_ddp.py \
-          --hpus 1 -m "not standalone_only" \
+          --hpus 1 -W ignore::FutureWarning -m "not standalone_only" \
           --junitxml=hpu_test-torch-results.xml
       displayName: 'HPU General tests'
 

diff --git a/tests/test_pytorch/strategies/test_fsdp.py b/tests/test_pytorch/strategies/test_fsdp.py
@@ -324,7 +324,7 @@ def training_step(self, batch, batch_idx):
     trainer.fit(model)
 
 
-@pytest.mark.parametrize("state_dict_type", [pytest.param("sharded", marks=pytest.mark.xfail(reason="Failure with 1.18")), "full"])  # noqa: E501
+@pytest.mark.parametrize("state_dict_type", ["sharded", "full"])
 @pytest.mark.standalone()
 @pytest.mark.skipif(get_device_name_from_hlsmi() == "GAUDI", reason="The tests requires Gaudi2 and above.")
 def test_fsdp_strategy_checkpoint(tmpdir, arg_hpus, state_dict_type):