From 850dceb80e8cb78a9eaeb0351118d57a32820fb9 Mon Sep 17 00:00:00 2001
From: Tim Lukas <tim.lukas-1@studium.uni-hamburg.de>
Date: Tue, 5 Aug 2025 10:10:26 +0200
Subject: [PATCH] Add Docker workflow for PyTorch image builds and pushes

---
 .github/workflows/docker-publish-pytorch.yml | 65 ++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 .github/workflows/docker-publish-pytorch.yml

diff --git a/.github/workflows/docker-publish-pytorch.yml b/.github/workflows/docker-publish-pytorch.yml
new file mode 100644
index 0000000..f0bac17
--- /dev/null
+++ b/.github/workflows/docker-publish-pytorch.yml
@@ -0,0 +1,65 @@
+name: Docker
+
+# This workflow uses actions that are not certified by GitHub.
+# They are provided by a third-party and are governed by
+# separate terms of service, privacy policy, and support
+# documentation.
+
+on:
+  push:
+    branches: [ "PyTorch" ]
+    # Publish semver tags as releases.
+    tags: [ 'v*.*.*' ]
+  pull_request:
+    branches: [ "PyTorch" ]
+
+env:
+  REGISTRY:
+  IMAGE_NAME: ${{ vars.DOCKERHUB_USERNAME }}/${{ vars.DOCKERHUB_REPO }}
+
+jobs:
+  build_latest:
+    runs-on: ubuntu-latest
+    steps:
+
+      - name: Checkout
+        uses: actions/checkout@main
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Build image (without tag for pull request)
+        if: github.event_name == 'pull_request'
+        run: docker build -f Dockerfile .
+
+      - name: Build image
+        if: github.event_name == 'push' && github.ref == 'refs/heads/PyTorch'
+        run: docker build -f Dockerfile --tag ${IMAGE_NAME}:pytorch .
+
+      - name: Login to Dockerhub container registry
+        if: github.event_name == 'push' && github.ref == 'refs/heads/PyTorch'
+        run: docker login -u ${{ vars.DOCKERHUB_USERNAME }} -p ${{ secrets.DOCKERHUB_TOKEN }}
+
+      - name: Push image
+        if: github.event_name == 'push' && github.ref == 'refs/heads/PyTorch'
+        run: docker push ${IMAGE_NAME}:pytorch
+
+  build_release:
+    # Build an extra image for tagged commits
+    runs-on: ubuntu-latest
+    if: startsWith(github.event.ref, 'refs/tags')
+    steps:
+      - name: Checkout
+        uses: actions/checkout@main
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Build image
+        run: docker build -f Dockerfile --tag ${IMAGE_NAME}:${{  github.ref_name  }} .
+
+      - name: Login to Dockerhub container registry
+        run: docker login -u ${{ vars.DOCKERHUB_USERNAME }} -p ${{ secrets.DOCKERHUB_TOKEN }}
+
+      - name: Push image
+        run: docker push ${IMAGE_NAME}:${{  github.ref_name  }}