prune_pytest_as_errors

Author: Borda

.actions/assistant.py

@@ -459,6 +459,25 @@ def generate_docker_tags(
         tags = [f"{docker_project}:{tag}" for tag in tags]
         print(",".join(tags))
 
+    @staticmethod
+    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
+
+        with open(pyproject_toml, encoding="utf-8") as fopen:
+            pyproject = toml.load(fopen)
+        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)
+
 
 if __name__ == "__main__":
     import jsonargparse

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

@@ -105,7 +105,7 @@ jobs:
           uv pip install "cython<3.0" wheel
           uv pip install "pyyaml==5.4" --no-build-isolation
           # This script removes any line containing "error::FutureWarning" from pyproject.toml
-          sed -i '/error::FutureWarning/d' pyproject.toml
+          python .actions/assistant.py prune_pytest_as_errors
 
       - name: Adjust PyTorch versions in requirements files
         if: ${{ matrix.config.requires != 'oldest' }}

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

@@ -104,7 +104,7 @@ jobs:
           uv pip install "cython<3.0" wheel
           uv pip install "pyyaml==5.4" --no-build-isolation
           # This script removes any line containing "error::FutureWarning" from pyproject.toml
-          sed -i '/error::FutureWarning/d' pyproject.toml
+          python .actions/assistant.py prune_pytest_as_errors
 
       - name: Adjust PyTorch versions in requirements files
         if: ${{ matrix.config.requires != 'oldest' }}