Work around type of yaml.safe_dump's width parameter

It works with inf, but doesn't officially support float. I'm not aware of another way to prevent line wrapping though.
dseomn · Sep 24, 2023 · 3394c20 · 3394c20
1 parent c0d33e5
commit 3394c20
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 5 deletions.
diff --git a/rock_paper_sand/config.py b/rock_paper_sand/config.py
@@ -16,6 +16,7 @@
 import dataclasses
 import difflib
 import functools
+import typing
 from typing import Any, Self
 
 from google.protobuf import json_format
@@ -97,10 +98,14 @@ def _lint_sort(self) -> Mapping[str, Any]:
             "sort": "".join(
                 difflib.unified_diff(
                     yaml.safe_dump(
-                        names, allow_unicode=True, width=float("inf")
+                        names,
+                        allow_unicode=True,
+                        width=typing.cast(int, float("inf")),
                     ).splitlines(keepends=True),
                     yaml.safe_dump(
-                        names_sorted, allow_unicode=True, width=float("inf")
+                        names_sorted,
+                        allow_unicode=True,
+                        width=typing.cast(int, float("inf")),
                     ).splitlines(keepends=True),
                     fromfile="media-names",
                     tofile="media-names-sorted",

diff --git a/rock_paper_sand/config_subcommand.py b/rock_paper_sand/config_subcommand.py
@@ -15,6 +15,7 @@
 
 import argparse
 import sys
+import typing
 
 import yaml
 
@@ -38,7 +39,7 @@ def run(self, args: argparse.Namespace):
                         default_style="|",
                         sort_keys=False,
                         allow_unicode=True,
-                        width=float("inf"),
+                        width=typing.cast(int, float("inf")),
                     ),
                     end="",
                 )

diff --git a/rock_paper_sand/report.py b/rock_paper_sand/report.py
@@ -18,6 +18,7 @@
 import email.message
 import json
 import subprocess
+import typing
 from typing import Any
 
 import yaml
@@ -67,7 +68,7 @@ def _dump_for_email(results: Any) -> str:
         results,
         sort_keys=False,
         allow_unicode=True,
-        width=float("inf"),
+        width=typing.cast(int, float("inf")),
     )
 
 

diff --git a/rock_paper_sand/reports_subcommand.py b/rock_paper_sand/reports_subcommand.py
@@ -14,6 +14,7 @@
 """Reports commands."""
 
 import argparse
+import typing
 
 import yaml
 
@@ -71,7 +72,7 @@ def run(self, args: argparse.Namespace):
                     results,
                     sort_keys=False,
                     allow_unicode=True,
-                    width=float("inf"),
+                    width=typing.cast(int, float("inf")),
                 ),
                 end="",
             )