Enable checkpoint tests

Lightning-AI · Nov 4, 2024 · 6a4eaaa · 6a4eaaa
1 parent 66f4485
commit 6a4eaaa
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.azure/hpu-tests.yml b/.azure/hpu-tests.yml
@@ -115,6 +115,7 @@ jobs:
           tests/test_pytorch/test_datamodule.py \
           tests/test_pytorch/test_profiler.py \
           tests/test_pytorch/test_precision.py \
+          tests/test_pytorch/test_checkpointing.py \
           tests/test_pytorch/strategies/test_hpu_parallel.py \
           tests/test_pytorch/strategies/test_hpu_ddp.py \
           --hpus 1 -W ignore::FutureWarning -m "not standalone_only" \
@@ -155,6 +156,7 @@ jobs:
         bash tests/run_standalone_tests.sh --hpus 1 -m standalone_only -f \
           tests/test_pytorch/strategies/test_hpu_parallel.py \
           tests/test_pytorch/test_precision.py \
+          tests/test_pytorch/test_checkpointing.py \
           tests/test_pytorch/test_dynamic_shapes.py
       displayName: Standalone-only single card tests