update

Author: Roman

.github/workflows/monitor_requirements_size_master.yml

@@ -7,7 +7,7 @@ name: Monitor SDK Requirements Size
 
 on:
   pull_request:
-    types: [opened]
+    types: [opened, synchronize]
     branches: [master]
 
 jobs:
@@ -16,61 +16,59 @@ jobs:
     strategy:
       matrix:
         python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
-
-    name: Python ${{ matrix.python-version }}
+    outputs:
+      py39: ${{ steps.set-output.outputs.py39 }}
+      py310: ${{ steps.set-output.outputs.py310 }}
+      py311: ${{ steps.set-output.outputs.py311 }}
+      py312: ${{ steps.set-output.outputs.py312 }}
+      py313: ${{ steps.set-output.outputs.py313 }}
     steps:
-      - name: Checkout code
-        uses: actions/checkout@v4
-
-      - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v5
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
 
-      - name: Create virtualenv and install Bittensor SDK
+      - name: Create virtualenv and install
         run: |
           python -m venv venv
           source venv/bin/activate
           pip install --upgrade pip
           pip install .
 
-      - name: Measure venv folder size
-        id: venv-size
-        run: |
-          du -sm venv | cut -f1 > venv_size_mb.txt
-          echo "venv_size=$(cat venv_size_mb.txt)" >> $GITHUB_OUTPUT
-
-      - name: Save size for summary
+      - name: Measure venv size
+        id: set-output
         run: |
-          echo "Python ${{ matrix.python-version }}: ${{ steps.venv-size.outputs.venv_size }} MB" >> $GITHUB_WORKSPACE/venv_sizes.txt
-
-      - name: Upload summary file as artifact
-        uses: actions/upload-artifact@v4
-        with:
-          name: venv_sizes
-          path: $GITHUB_WORKSPACE/venv_sizes.txt
+          SIZE=$(du -sm venv | cut -f1)
+          VERSION=${{ matrix.python-version }}
+          echo "Detected size: $SIZE MB for Python $VERSION"
+          case "$VERSION" in
+            3.9)  echo "py39=$SIZE" >> $GITHUB_OUTPUT ;;
+            3.10) echo "py310=$SIZE" >> $GITHUB_OUTPUT ;;
+            3.11) echo "py311=$SIZE" >> $GITHUB_OUTPUT ;;
+            3.12) echo "py312=$SIZE" >> $GITHUB_OUTPUT ;;
+            3.13) echo "py313=$SIZE" >> $GITHUB_OUTPUT ;;
+          esac
 
   comment-on-pr:
     if: github.event_name == 'pull_request' && github.base_ref == 'master'
-    runs-on: ubuntu-latest
     needs: measure-venv
-
+    runs-on: ubuntu-latest
     steps:
-      - name: Download summary
-        uses: actions/download-artifact@v4
-        with:
-          name: venv_sizes
-          path: .
-
-      - name: Read summary and comment it
+      - name: Comment on PR
         uses: actions/github-script@v7
         with:
           script: |
-            const fs = require('fs');
-            const comment = fs.readFileSync('venv_sizes.txt', 'utf8');
+            const sizes = {
+              "3.9": "${{ needs.measure-venv.outputs.py39 || 'N/A' }}",
+              "3.10": "${{ needs.measure-venv.outputs.py310 || 'N/A' }}",
+              "3.11": "${{ needs.measure-venv.outputs.py311 || 'N/A' }}",
+              "3.12": "${{ needs.measure-venv.outputs.py312 || 'N/A' }}",
+              "3.13": "${{ needs.measure-venv.outputs.py313 || 'N/A' }}"
+            };
+            const lines = Object.entries(sizes).map(([ver, size]) => `Python ${ver}: ${size} MB`);
             github.rest.issues.createComment({
               issue_number: context.payload.pull_request.number,
               owner: context.repo.owner,
               repo: context.repo.name,
-              body: `Bittensor SDK virtual environment sizes by Python version:**\n\n\`\`\`\n${comment}\n\`\`\``
+              body: `Bittensor SDK virtual environment sizes by Python version:**\n\n\`\`\`\n${lines.join("\n")}\n\`\`\``
             });