From db88420014f4e7968ec1373c4fcc569ac38716ca Mon Sep 17 00:00:00 2001
From: Sander Land <48946947+sanderland@users.noreply.github.com>
Date: Sun, 15 Jun 2025 14:23:11 +0200
Subject: [PATCH] tomllib (#766)

---
 pyproject.toml        | 1 +
 tests/test_version.py | 6 +++++-
 uv.lock               | 2 ++
 3 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 5eeb233..7e63127 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -36,6 +36,7 @@ dev = [
     "polib>=1.2.0,<2",
     "pyinstaller>=6.14.1",
     "pytest>=8.3.2,<9",
+    "tomli>=1.2.0 ; python_version < '3.11'",
 ]
 
 [tool.uv]
diff --git a/tests/test_version.py b/tests/test_version.py
index 7f2512f..f0a310b 100644
--- a/tests/test_version.py
+++ b/tests/test_version.py
@@ -1,4 +1,8 @@
-import tomllib
+try:
+    import tomllib  # Python 3.11+
+except ImportError:
+    import tomli as tomllib  # Python < 3.11
+
 from pathlib import Path
 
 from katrain.core.constants import VERSION
diff --git a/uv.lock b/uv.lock
index c6f416f..74e67d0 100644
--- a/uv.lock
+++ b/uv.lock
@@ -281,6 +281,7 @@ dev = [
     { name = "polib" },
     { name = "pyinstaller" },
     { name = "pytest" },
+    { name = "tomli", marker = "python_full_version < '3.11'" },
 ]
 
 [package.metadata]
@@ -301,6 +302,7 @@ dev = [
     { name = "polib", specifier = ">=1.2.0,<2" },
     { name = "pyinstaller", specifier = ">=6.14.1" },
     { name = "pytest", specifier = ">=8.3.2,<9" },
+    { name = "tomli", marker = "python_full_version < '3.11'", specifier = ">=1.2.0" },
 ]
 
 [[package]]