ci: Working on schedulers

jupyter-naas · Jul 31, 2024 · 4f7f262 · 4f7f262
1 parent 3e5f91e
commit 4f7f262
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/.github/scripts/generate_schedulers.py b/.github/scripts/generate_schedulers.py
@@ -22,11 +22,11 @@
           password: ${{ secrets.GITHUB_TOKEN }}
 
       - name: Pull Docker image
-        run: docker pull ghcr.io/jupyter-naas/abi/abi:latest
+        run: docker pull ghcr.io/${{ github.repository }}/abi:latest
 
       # - name: Run Papermill
       #   run: |
-      #     docker run --name abi-execution -i --platform linux/amd64 ghcr.io/jupyter-naas/abi/abi:latest ls
+      #     docker run --name abi-execution -i --platform linux/amd64 ghcr.io/${{ github.repository }}/abi:latest ls
       #     mkdir output
       #     docker cp abi-execution:/app/__pipeline__.ipynb ./output/__pipeline__.ipynb
 
@@ -62,7 +62,7 @@ def generate_schedulers(config : dict, template : str):
 export SCHEDULER_ID=$(python -c "import uuid; print(uuid.uuid4())")
 
 # Execute the Scheduler script
-docker run --name $SCHEDULER_ID -i --platform linux/amd64 ghcr.io/jupyter-naas/abi/abi:latest python .github/scripts/run_scheduler.py "{scheduler['name']}"
+docker run --name $SCHEDULER_ID -i --platform linux/amd64 ghcr.io/${{ github.repository }}/abi:latest python .github/scripts/run_scheduler.py "{scheduler['name']}"
 
 # Create the output directory that will be used to store the output files and save them as artifacts.
 mkdir -p outputs/

diff --git a/.github/workflows/scheduler__main.yaml b/.github/workflows/scheduler__main.yaml
@@ -11,7 +11,7 @@ jobs:
         registry: ghcr.io
         username: ${{ github.actor }}
     - name: Pull Docker image
-      run: docker pull ghcr.io/jupyter-naas/abi/abi:latest
+      run: docker pull ghcr.io/${{ github.repository }}/abi:latest
     - name: main
       run: '
 
@@ -22,8 +22,8 @@ jobs:
 
         # Execute the Scheduler script
 
-        docker run --name $SCHEDULER_ID -i --platform linux/amd64 ghcr.io/jupyter-naas/abi/abi:latest
-        python .github/scripts/run_scheduler.py "main"
+        docker run --name $SCHEDULER_ID -i --platform linux/amd64 ghcr.io/${ github.repository
+        }/abi:latest python .github/scripts/run_scheduler.py "main"
 
 
         # Create the output directory that will be used to store the output files