update

Author: Borda

.actions/assistant.py

@@ -464,19 +464,19 @@ def prune_pytest_as_errors(
         pyproject_toml: str = "pyproject.toml", errors: tuple = ("FutureWarning", "DeprecationWarning")
     ) -> None:
         """Prune pytest warnings as errors from the pyproject.toml file."""
-        import toml
+        import tomlkit
 
         with open(pyproject_toml, encoding="utf-8") as fopen:
-            pyproject = toml.load(fopen)
+            content = fopen.read()
+        pyproject = tomlkit.parse(content)
         filterwarnings = pyproject.get("tool", {}).get("pytest", {}).get("ini_options", {}).get("filterwarnings", [])
         if not filterwarnings:
             return
-        print(filterwarnings)
-        print([f"error::{err}" for err in errors])
         filterwarnings = [wrn for wrn in filterwarnings if not any(f"error::{err}" in wrn for err in errors)]
         pyproject["tool"]["pytest"]["ini_options"]["filterwarnings"] = filterwarnings
+
         with open(pyproject_toml, "w", encoding="utf-8") as fopen:
-            toml.dump(pyproject, fopen)
+            fopen.write(tomlkit.dumps(pyproject))
 
 
 if __name__ == "__main__":

.github/workflows/ci-tests-fabric.yml

@@ -106,7 +106,6 @@ jobs:
           uv pip install "pyyaml==5.4" --no-build-isolation
           # This script removes any line containing "error::FutureWarning" from pyproject.toml
           uv pip install -r requirements/ci.txt
-          uv pip list # debugging
           python .actions/assistant.py prune_pytest_as_errors
 
       - name: Adjust PyTorch versions in requirements files

.github/workflows/ci-tests-pytorch.yml

@@ -105,7 +105,6 @@ jobs:
           uv pip install "pyyaml==5.4" --no-build-isolation
           # This script removes any line containing "error::FutureWarning" from pyproject.toml
           uv pip install -r requirements/ci.txt
-          uv pip list # debugging
           python .actions/assistant.py prune_pytest_as_errors
 
       - name: Adjust PyTorch versions in requirements files