Move from tomllib to tomli (#3714)

microsoft · Oct 9, 2024 · 8e50e58 · 8e50e58
1 parent 23fd92a
commit 8e50e58
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 160 deletions.
diff --git a/python/pyproject.toml b/python/pyproject.toml
@@ -19,6 +19,7 @@ dev-dependencies = [
     "mypy-protobuf",
     "cookiecutter",
     "poethepoet",
+    "tomli",
 ]
 
 [tool.uv.sources]

diff --git a/python/run_task_in_pkgs_if_exist.py b/python/run_task_in_pkgs_if_exist.py
@@ -3,14 +3,14 @@
 from pathlib import Path
 from typing import List
 
-import tomllib
+import tomli
 from poethepoet.app import PoeThePoet
 from rich import print
 
 
 def discover_projects(workspace_pyproject_file: Path) -> List[Path]:
     with workspace_pyproject_file.open("rb") as f:
-        data = tomllib.load(f)
+        data = tomli.load(f)
 
     projects = data["tool"]["uv"]["workspace"]["members"]
 
@@ -28,7 +28,7 @@ def discover_projects(workspace_pyproject_file: Path) -> List[Path]:
 
 def extract_poe_tasks(file: Path) -> set[str]:
     with file.open("rb") as f:
-        data = tomllib.load(f)
+        data = tomli.load(f)
 
     tasks = set(data.get("tool", {}).get("poe", {}).get("tasks", {}).keys())