Skip MemPoolTest on github workflow

Differential Revision: D88655756

Author: tanquer

.github/workflows/build_test.yaml

@@ -62,5 +62,5 @@ jobs:
         python -c "import torchcomms; import torchcomms._transport"
 
         # Run tests
-        comms/torchcomms/scripts/run_tests_integration_py.sh
+        comms/torchcomms/scripts/run_tests_integration_py.sh -s MemPoolTest.py
         pytest -v comms/torchcomms/tests/unit/py

comms/torchcomms/scripts/run_tests_integration_py.sh

@@ -9,9 +9,45 @@ set -ex
 
 cd "$(dirname "$0")/../tests/integration/py"
 
+SKIP_TESTS=""
+
+# Function to display usage
+usage() {
+    echo "Usage: $0 [options]"
+    echo "Options:"
+    echo "  --skips, -s         Comma-separated list of tests to skip"
+    exit 1
+}
+
+# Parse command line arguments
+while [[ $# -gt 0 ]]; do
+    key="$1"
+    case $key in
+        --skip|-s)
+            SKIP_TESTS="$2"
+            shift 2
+            ;;
+        *)
+            echo "Unknown option: $1"
+            usage
+            ;;
+    esac
+done
+
 run_tests () {
+    # Convert comma-separated skip tests to array
+    IFS=',' read -ra SKIP_TESTS_ARRAY <<< "$SKIP_TESTS"
     for file in *Test.py; do
-        torchrun --nnodes 1 --nproc_per_node 4 "$file" --verbose
+        skip=false
+        for skip_file in "${SKIP_TESTS_ARRAY[@]}"; do
+            if [[ "$file" == "$skip_file" ]]; then
+                skip=true
+                break
+            fi
+        done
+        if ! $skip; then
+            torchrun --nnodes 1 --nproc_per_node 4 "$file" --verbose
+        fi
     done
 }