Updated release file.

Author: vbharadwaj-bk

.github/workflows/release.yaml

@@ -6,7 +6,7 @@ on:
 # ref: https://packaging.python.org/en/latest/guides/publishing-package-distribution-releases-using-github-actions-ci-cd-workflows/
 
 jobs:
-  build:
+  build-oeq:
     name: Build distribution
     runs-on: ubuntu-latest
     steps:
@@ -16,20 +16,21 @@ jobs:
       with:
         python-version: '3.10'
     - name: install dependencies, then build source tarball
-      run: | 
+      run: |
+        cd openequivariance 
         python3 -m pip install build --user
         python3 -m build --sdist 
     - name: store the distribution packages 
       uses: actions/upload-artifact@v4
       with:
         name: python-package-distributions
-        path: dist/
+        path: openequivariance/dist/
 
   pypi-publish:
     name: Upload release to PyPI
     runs-on: ubuntu-latest
     # build task to be completed first
-    needs: build
+    needs: build-oeq
     # Specifying a GitHub environment is optional, but strongly encouraged
     environment: 
         name: pypi
@@ -42,6 +43,49 @@ jobs:
         uses: actions/download-artifact@v4
         with:
           name: python-package-distributions
-          path: dist/
+          path: openequivariance/dist/
+      - name: publish package distributions to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+
+  # ------------------------------------
+
+  build-oeq-extjax:
+    name: Build distribution
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v4
+    - name: set up Python
+      uses: actions/setup-python@v5
+      with:
+        python-version: '3.10'
+    - name: install dependencies, then build source tarball
+      run: |
+        cd openequivariance_extjax 
+        python3 -m pip install build --user
+        python3 -m build --sdist 
+    - name: store the distribution packages 
+      uses: actions/upload-artifact@v4
+      with:
+        name: python-package-distributions
+        path: openequivariance_extjax/dist/
+
+  pypi-publish:
+    name: Upload release to PyPI
+    runs-on: ubuntu-latest
+    # build task to be completed first
+    needs: build-oeq-extjax
+    # Specifying a GitHub environment is optional, but strongly encouraged
+    environment: 
+        name: pypi
+        url: https://pypi.org/p/openequivariance_extjax
+    permissions:
+      # IMPORTANT: this permission is mandatory for Trusted Publishing
+      id-token: write
+    steps:
+      - name: download the distributions
+        uses: actions/download-artifact@v4
+        with:
+          name: python-package-distributions
+          path: openequivariance_extjax/dist/
       - name: publish package distributions to PyPI
         uses: pypa/gh-action-pypi-publish@release/v1