Revert "fix payload in workflows (#331)"

This reverts commit 015607b.

Author: msaroufim

.github/workflows/amd_workflow.yml

@@ -39,8 +39,8 @@ jobs:
         # Apply mask to the extracted content
         echo "::add-mask::$PAYLOAD"
         
-        # Compress and base64 encode the payload to match what the Python script expects
-        echo "$PAYLOAD" | python3 -c "import sys, zlib, base64; print(base64.b64encode(zlib.compress(sys.stdin.read().encode('utf-8'))).decode('ascii'))" > payload.json
+        # Now write to file (won't be logged since it's masked)
+        echo "$PAYLOAD" > payload.json
     
     - name: Set venv directory based on runner
       run: |

.github/workflows/nvidia_workflow.yml

@@ -46,8 +46,8 @@ jobs:
         # Apply mask to the extracted content
         echo "::add-mask::$PAYLOAD"
         
-        # Compress and base64 encode the payload to match what the Python script expects
-        echo "$PAYLOAD" | python3 -c "import sys, zlib, base64; print(base64.b64encode(zlib.compress(sys.stdin.read().encode('utf-8'))).decode('ascii'))" > payload.json
+        # Now write to file (won't be logged since it's masked)
+        echo "$PAYLOAD" > payload.json
 
     - name: Install uv
       uses: astral-sh/setup-uv@v3

scripts/github_test_payload.json

@@ -4,5 +4,5 @@
     "main.py": "import torch\n\nprint(f\"CUDA available: {torch.cuda.is_available()}\")\nprint(f\"PyTorch version: {torch.__version__}\")\n\nif torch.cuda.is_available():\n  x = torch.randn(5, device='cuda')\n  print(f\"Random tensor on GPU: {x}\")"
   },
   "main": "main.py",
-  "mode": "test"
+  "mode": "script"
 }