Update pypi release action.

The package name is `bayeux`, but the project name on pypi is `bayeux-ml`.

PiperOrigin-RevId: 592551676

.github/workflows/pytest_and_autopublish.yml

@@ -52,4 +52,3 @@ jobs:
         pypi-token: ${{ secrets.PYPI_API_TOKEN }}
         gh-token: ${{ secrets.GITHUB_TOKEN }}
         parse-changelog: true
-        pkg-name: bayeux_ml

pyproject.toml

@@ -1,5 +1,5 @@
 [project]
-name = "bayeux"
+name = "bayeux-ml"
 description = "Stitching together probabilistic models and inference."
 readme = "README.md"
 requires-python = ">=3.9"
@@ -33,6 +33,11 @@ repository = "https://github.com/jax-ml/bayeux"
 changelog = "https://github.com/jax-ml/bayeux/blob/main/CHANGELOG.md"
 # documentation = ""
 
+[tool.poetry]
+packages = [
+    { include = "bayeux" },
+]
+
 [project.optional-dependencies]
 # Development deps (unittest, linting, formating,...)
 # Installed through `pip install -e .[dev]`
@@ -56,6 +61,9 @@ pyink-use-majority-quotes = true
 requires = ["flit_core >=3.8,<4"]
 build-backend = "flit_core.buildapi"
 
+[tool.flit.module]
+name = "bayeux"
+
 [tool.flit.sdist]
 # Flit specific options (files to exclude from the PyPI package).
 # If using another build backend (setuptools, poetry), you can remove this