Update pyproject.toml to be PEP 621 compliant

PiperOrigin-RevId: 689745799

gemma/__init__.py

@@ -26,3 +26,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ============================================================================
+
+"""gemma API."""
+
+# A new PyPI release will be pushed every time `__version__` is increased.
+# When changing this, also update the CHANGELOG.md.
+__version__ = '1.0.0'

pyproject.toml

@@ -1,37 +1,80 @@
-[tool.poetry]
+[project]
+# Project metadata. Available keys are documented at:
+# https://packaging.python.org/en/latest/specifications/declaring-project-metadata
 name = "gemma"
-version = "1.0.0"
 description = "Open weights large language model (LLM) from Google DeepMind."
+readme = "README.md"
+requires-python = ">=3.10"
+license = {file = "LICENSE"}
 authors = [
-  "Alek Andreev <[email protected]>",
-  "Armand Joulin <[email protected]>",
-  "Cassidy Hardin <[email protected]>",
-  "Juliette Love <[email protected]>",
-  "Kathleen Kenealy <[email protected]>",
-  "Laurent Sifre <[email protected]>",
-  "Léonard Hussenot <[email protected]>",
-  "Michelle Casbon <[email protected]>",
-  "Mihir Sanjay Kale <[email protected]>",
-  "Morgane Riviere <[email protected]>",
-  "Ravin Kumar <[email protected]>",
-  "Robert Dadashi <[email protected]>",
-  "Shreya Pathak <[email protected]>",
-  "Surya Bhupatiraju <[email protected]>",
-  "Thomas Mesnard <[email protected]>"
+    {name = "Alek Andreev", email="[email protected]"},
+    {name = "Armand Joulin", email="[email protected]"},
+    {name = "Cassidy Hardin", email="[email protected]"},
+    {name = "Juliette Love", email="[email protected]"},
+    {name = "Kathleen Kenealy", email="[email protected]"},
+    {name = "Laurent Sifre", email="[email protected]"},
+    {name = "Léonard Hussenot", email="[email protected]"},
+    {name = "Michelle Casbon", email="[email protected]"},
+    {name = "Mihir Sanjay Kale", email="[email protected]"},
+    {name = "Morgane Riviere", email="[email protected]"},
+    {name = "Ravin Kumar", email="[email protected]"},
+    {name = "Robert Dadashi", email="[email protected]"},
+    {name = "Shreya Pathak", email="[email protected]"},
+    {name = "Surya Bhupatiraju", email="[email protected]"},
+    {name = "Thomas Mesnard", email="[email protected]"},
+]
+classifiers = [  # List of https://pypi.org/classifiers/
+    "License :: OSI Approved :: Apache Software License",
+    "Intended Audience :: Science/Research",
+]
+keywords = []
+
+# pip dependencies of the project
+# Installed locally with `pip install -e .`
+dependencies = [
+    "absl-py",
+    "sentencepiece",
+    "flax",
 ]
-python-version = "3.10"
-license = "Apache-2.0"
-readme = "README.md"
 
-[tool.poetry.dependencies]
-absl-py = "^2.1.0"
-sentencepiece = "^0.1.99"
-flax = ">=0.8"
-pytest = {version = "^8.0.0", optional = true}
+# `version` is automatically set by flit to use `gemma.__version__`
+dynamic = ["version"]
 
-[tool.poetry.extras]
-test = ["pytest"]
+[project.urls]
+homepage = "https://github.com/google-deepmind/gemma"
+repository = "https://github.com/google-deepmind/gemma"
+changelog = "https://github.com/google-deepmind/gemma/blob/main/CHANGELOG.md"
+# documentation = ""
+
+[project.optional-dependencies]
+# Development deps (unittest, linting, formating,...)
+# Installed through `pip install -e .[dev]`
+dev = [
+    "pytest",
+    "pytest-xdist",
+    "pylint>=2.6.0",
+    "pyink",
+]
+
+[tool.pyink]
+# Formatting configuration to follow Google style-guide
+line-length = 80
+unstable = true
+pyink-indentation = 2
+pyink-use-majority-quotes = true
 
 [build-system]
-requires = ["poetry-core"]
-build-backend = "poetry.core.masonry.api"
+# Build system specify which backend is used to build/install the project (flit,
+# poetry, setuptools,...). All backends are supported by `pip install`
+requires = ["flit_core >=3.8,<4"]
+build-backend = "flit_core.buildapi"
+
+[tool.flit.sdist]
+# Flit specific options (files to exclude from the PyPI package).
+# If using another build backend (setuptools, poetry), you can remove this
+# section.
+exclude = [
+  # Do not release tests files on PyPI
+  "**/*_test.py",
+  "testdata/**",
+]