DEBUG - skip to standalone

Author: carmocca

.azure/gpu-tests.yml

@@ -125,21 +125,14 @@ jobs:
         python requirements/pytorch/check-avail-extras.py
       displayName: 'Env details'
 
-    - bash: bash .actions/pull_legacy_checkpoints.sh
-      displayName: 'Get legacy checkpoints'
-
-    - bash: python -m coverage run --source pytorch_lightning -m pytest .
-      workingDirectory: src/pytorch_lightning
-      displayName: 'Testing: PyTorch doctests'
-
-    - bash: python -m coverage run --source pytorch_lightning -m pytest --ignore benchmarks -v --junitxml=$(Build.StagingDirectory)/test-results.xml --durations=50
+    - bash: pytest tests/tests_pytorch/strategies/test_bagua_strategy.py
       env:
+        PL_USE_MOCKED_MNIST: "1"
         PL_RUN_CUDA_TESTS: "1"
-      workingDirectory: tests/tests_pytorch
-      displayName: 'Testing: PyTorch standard'
+      displayName: 'ASD'
       timeoutInMinutes: "35"
 
-    - bash: bash run_standalone_tests.sh
+    - bash: bash run_standalone_tests.sh -k bagua
       workingDirectory: tests/tests_pytorch
       env:
         PL_USE_MOCKED_MNIST: "1"

tests/tests_pytorch/run_standalone_tests.sh

@@ -17,13 +17,13 @@ set -e
 
 # Batch size for testing: Determines how many standalone test invocations run in parallel
 # It can be set through the env variable PL_STANDALONE_TESTS_BATCH_SIZE and defaults to 6 if not set
-test_batch_size="${PL_STANDALONE_TESTS_BATCH_SIZE:-6}"
+test_batch_size="${PL_STANDALONE_TESTS_BATCH_SIZE:-1}"
 source="${PL_STANDALONE_TESTS_SOURCE}"
 
 # this environment variable allows special tests to run
 export PL_RUN_STANDALONE_TESTS=1
 # python arguments
-defaults="-m coverage run --source $source --append -m pytest --no-header"
+defaults="-m pytest --no-header"
 
 # find tests marked as `@RunIf(standalone=True)`. done manually instead of with pytest because it is faster
 grep_output=$(grep --recursive --word-regexp . --regexp 'standalone=True' --include '*.py')