Support Python 3.10 TOML loading (#626)

Author: cvanelteren

tools/release/publish_zenodo.py

@@ -6,10 +6,14 @@
 import mimetypes
 import os
 import sys
-import tomllib
 from pathlib import Path
 from urllib import error, parse, request
 
+try:
+    import tomllib
+except ModuleNotFoundError:  # pragma: no cover
+    import tomli as tomllib
+
 try:
     import yaml
 except ImportError as exc:  # pragma: no cover - exercised in release workflow

ultraplot/tests/test_release_metadata.py

@@ -3,7 +3,6 @@
 import importlib.util
 import re
 import subprocess
-import tomllib
 from pathlib import Path
 
 import pytest
@@ -87,8 +86,7 @@ def test_zenodo_release_metadata_is_built_from_repository_sources():
     """
     publish_zenodo = _load_publish_zenodo()
     citation = yaml.safe_load(CITATION_CFF.read_text(encoding="utf-8"))
-    with PYPROJECT.open("rb") as handle:
-        pyproject = tomllib.load(handle)
+    pyproject = publish_zenodo.load_pyproject(PYPROJECT)
     metadata = publish_zenodo.build_metadata(citation, pyproject)
     assert metadata["title"] == citation["title"]
     assert metadata["upload_type"] == "software"