Use tomllib

gediminasz · Jun 30, 2023 · 6f80b57 · 6f80b57
1 parent c6cd2c8
commit 6f80b57
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/misc/test_version.py b/tests/misc/test_version.py
@@ -1,9 +1,12 @@
-import tomli
+try:
+    import tomllib
+except ImportError:  # TODO remove this workaround when Python 3.10 support is dropped
+    import tomli as tomllib
 
 import slskit
 
 
 def test_version_in_module_matches_pyproject():
     with open("pyproject.toml", "rb") as project_file:
-        project = tomli.load(project_file)
+        project = tomllib.load(project_file)
     assert slskit.__version__ == project["tool"]["poetry"]["version"]