From 97417f68b777c6e70ce8349525fb53d0117bccd9 Mon Sep 17 00:00:00 2001
From: SvenKlaassen <sven.klaassen@uni-hamburg.de>
Date: Fri, 21 Nov 2025 15:04:46 +0100
Subject: [PATCH]  improve deployment workflow for package publishing

---
 .github/workflows/deploy_pkg.yml | 81 ++++++++++++++++++++++++--------
 1 file changed, 61 insertions(+), 20 deletions(-)

diff --git a/.github/workflows/deploy_pkg.yml b/.github/workflows/deploy_pkg.yml
index 6c47470c..f104e4ca 100644
--- a/.github/workflows/deploy_pkg.yml
+++ b/.github/workflows/deploy_pkg.yml
@@ -5,33 +5,74 @@ on:
     types:
       - published
 
-
 jobs:
   build:
-
+    name: Build distribution 📦
     runs-on: ubuntu-latest
 
     steps:
-    - uses: actions/checkout@v4
-      with:
-        persist-credentials: false
+      - uses: actions/checkout@v4
+        with:
+          persist-credentials: false
+
+      - name: Install python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+
+      - name: Install build tools
+        run: |
+          python -m pip install --upgrade pip build twine
+
+      - name: Build package
+        run: python -m build
+
+      - name: Check package metadata
+        run: twine check dist/*
 
-    - name: Install python
-      uses: actions/setup-python@v5
-      with:
-        python-version: '3.12'
+      - uses: actions/upload-artifact@v4
+        with:
+          name: DoubleML-pkg
+          path: dist/
 
-    - name: Install build tools
-      run: |
-        python -m pip install --upgrade pip build twine
+  publish-to-testpypi:
+    name: Publish to TestPyPI 🧪
+    needs: build
+    runs-on: ubuntu-latest
+    environment:
+      name: testpypi
+      url: https://test.pypi.org/p/DoubleML
+    permissions:
+      id-token: write
+
+    steps:
+      - name: Download built package
+        uses: actions/download-artifact@v4
+        with:
+          name: DoubleML-pkg
+          path: dist/
 
-    - name: Build package
-      run: python -m build
+      - name: Publish to TestPyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          repository-url: https://test.pypi.org/legacy/
 
-    - name: Check package metadata
-      run: twine check dist/*
+  publish-to-pypi:
+    name: Publish to PyPI 🚀
+    needs: publish-to-testpypi
+    runs-on: ubuntu-latest
+    environment:
+      name: pypi
+      url: https://pypi.org/p/DoubleML
+    permissions:
+      id-token: write
+
+    steps:
+      - name: Download built package
+        uses: actions/download-artifact@v4
+        with:
+          name: DoubleML-pkg
+          path: dist/
 
-    - uses: actions/upload-artifact@v4
-      with:
-        name: DoubleML-pkg
-        path: dist/
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1