PyYAML: Use TypeAlias to mimic "infinity literal" restriction

#8973 (review)

Author: kmurphy4

stubs/PyYAML/yaml/__init__.pyi

@@ -46,7 +46,7 @@ def emit(
     Dumper=...,
     canonical: bool | None = ...,
     indent: int | None = ...,
-    width: float | None = ...,
+    width: int | _Inf | None = ...,
     allow_unicode: bool | None = ...,
     line_break: str | None = ...,
 ): ...
@@ -57,7 +57,7 @@ def serialize_all(
     Dumper=...,
     canonical: bool | None = ...,
     indent: int | None = ...,
-    width: float | None = ...,
+    width: int | _Inf | None = ...,
     allow_unicode: bool | None = ...,
     line_break: str | None = ...,
     encoding: str | None = ...,
@@ -73,7 +73,7 @@ def serialize_all(
     Dumper=...,
     canonical: bool | None = ...,
     indent: int | None = ...,
-    width: float | None = ...,
+    width: int | _Inf | None = ...,
     allow_unicode: bool | None = ...,
     line_break: str | None = ...,
     encoding: str | None = ...,
@@ -90,7 +90,7 @@ def serialize(
     *,
     canonical: bool | None = ...,
     indent: int | None = ...,
-    width: float | None = ...,
+    width: int | _Inf | None = ...,
     allow_unicode: bool | None = ...,
     line_break: str | None = ...,
     encoding: str | None = ...,
@@ -107,7 +107,7 @@ def serialize(
     *,
     canonical: bool | None = ...,
     indent: int | None = ...,
-    width: float | None = ...,
+    width: int | _Inf | None = ...,
     allow_unicode: bool | None = ...,
     line_break: str | None = ...,
     encoding: str | None = ...,
@@ -125,7 +125,7 @@ def dump_all(
     default_flow_style: bool | None = ...,
     canonical: bool | None = ...,
     indent: int | None = ...,
-    width: float | None = ...,
+    width: int | _Inf | None = ...,
     allow_unicode: bool | None = ...,
     line_break: str | None = ...,
     encoding: str | None = ...,
@@ -144,7 +144,7 @@ def dump_all(
     default_flow_style: bool | None = ...,
     canonical: bool | None = ...,
     indent: int | None = ...,
-    width: float | None = ...,
+    width: int | _Inf | None = ...,
     allow_unicode: bool | None = ...,
     line_break: str | None = ...,
     encoding: str | None = ...,
@@ -164,7 +164,7 @@ def dump(
     default_flow_style: bool | None = ...,
     canonical: bool | None = ...,
     indent: int | None = ...,
-    width: float | None = ...,
+    width: int | _Inf | None = ...,
     allow_unicode: bool | None = ...,
     line_break: str | None = ...,
     encoding: str | None = ...,
@@ -184,7 +184,7 @@ def dump(
     default_flow_style: bool | None = ...,
     canonical: bool | None = ...,
     indent: int | None = ...,
-    width: float | None = ...,
+    width: int | _Inf | None = ...,
     allow_unicode: bool | None = ...,
     line_break: str | None = ...,
     encoding: str | None = ...,
@@ -203,7 +203,7 @@ def safe_dump_all(
     default_flow_style: bool | None = ...,
     canonical: bool | None = ...,
     indent: int | None = ...,
-    width: float | None = ...,
+    width: int | _Inf | None = ...,
     allow_unicode: bool | None = ...,
     line_break: str | None = ...,
     encoding: str | None = ...,
@@ -222,7 +222,7 @@ def safe_dump_all(
     default_flow_style: bool | None = ...,
     canonical: bool | None = ...,
     indent: int | None = ...,
-    width: float | None = ...,
+    width: int | _Inf | None = ...,
     allow_unicode: bool | None = ...,
     line_break: str | None = ...,
     encoding: str | None = ...,
@@ -241,7 +241,7 @@ def safe_dump(
     default_flow_style: bool | None = ...,
     canonical: bool | None = ...,
     indent: int | None = ...,
-    width: float | None = ...,
+    width: int | _Inf | None = ...,
     allow_unicode: bool | None = ...,
     line_break: str | None = ...,
     encoding: str | None = ...,
@@ -260,7 +260,7 @@ def safe_dump(
     default_flow_style: bool | None = ...,
     canonical: bool | None = ...,
     indent: int | None = ...,
-    width: float | None = ...,
+    width: int | _Inf | None = ...,
     allow_unicode: bool | None = ...,
     line_break: str | None = ...,
     encoding: str | None = ...,

stubs/PyYAML/yaml/dumper.pyi

@@ -1,5 +1,6 @@
 from collections.abc import Mapping
 from typing import Any
+from typing_extensions import TypeAlias
 
 from yaml.emitter import Emitter
 from yaml.representer import BaseRepresenter, Representer, SafeRepresenter
@@ -8,6 +9,10 @@ from yaml.serializer import Serializer
 
 from .emitter import _WriteStream
 
+# Ideally, there would be a way to limit these values to only +/- float("inf"),
+# but that's not possible at the moment (https://github.com/python/typing/issues/1160).
+_Inf: TypeAlias = float
+
 class BaseDumper(Emitter, Serializer, BaseRepresenter, BaseResolver):
     def __init__(
         self,
@@ -16,7 +21,7 @@ class BaseDumper(Emitter, Serializer, BaseRepresenter, BaseResolver):
         default_flow_style: bool | None = ...,
         canonical: bool | None = ...,
         indent: int | None = ...,
-        width: float | None = ...,
+        width: int | _Inf | None = ...,
         allow_unicode: bool | None = ...,
         line_break: str | None = ...,
         encoding: str | None = ...,
@@ -35,7 +40,7 @@ class SafeDumper(Emitter, Serializer, SafeRepresenter, Resolver):
         default_flow_style: bool | None = ...,
         canonical: bool | None = ...,
         indent: int | None = ...,
-        width: float | None = ...,
+        width: int | _Inf | None = ...,
         allow_unicode: bool | None = ...,
         line_break: str | None = ...,
         encoding: str | None = ...,
@@ -54,7 +59,7 @@ class Dumper(Emitter, Serializer, Representer, Resolver):
         default_flow_style: bool | None = ...,
         canonical: bool | None = ...,
         indent: int | None = ...,
-        width: float | None = ...,
+        width: int | _Inf | None = ...,
         allow_unicode: bool | None = ...,
         line_break: str | None = ...,
         encoding: str | None = ...,