improve reusable workflow

Author: basfroman

.github/workflows/_run-e2e-single.yaml

@@ -1,3 +1,6 @@
+# WARNING: This file is used by multiple workflows. Please modify with caution. Any changes here may affect multiple
+# CI/CD pipelines.
+
 name: Run Single E2E Test with multiple Python versions
 
 on:
@@ -13,6 +16,16 @@ on:
         required: true
         type: string
         description: JSON array of Python versions
+      ref:
+        required: false
+        type: string
+        description: Git ref to checkout
+        default: ""
+      artifact-name:
+        required: false
+        type: string
+        description: Docker image artifact name
+        default: "subtensor-localnet"
 
 jobs:
   run-e2e:
@@ -28,6 +41,8 @@ jobs:
     steps:
       - name: Check-out repository
         uses: actions/checkout@v4
+        with:
+          ref: ${{ inputs.ref != '' && inputs.ref || github.ref }}
 
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v5
@@ -57,10 +72,10 @@ jobs:
       - name: Download Cached Docker Image
         uses: actions/download-artifact@v4
         with:
-          name: subtensor-localnet
+          name: ${{ inputs.artifact-name }}
 
       - name: Load Docker Image
-        run: docker load -i subtensor-localnet.tar
+        run: docker load -i ${{ inputs.artifact-name }}.tar
 
       - name: Run test with retry
         env: