fill_value

Author: crusaderky

array_api_compat/common/_aliases.py

@@ -67,7 +67,7 @@ def eye(
 
 def full(
     shape: Union[int, Tuple[int, ...]],
-    fill_value: Union[int, float],
+    fill_value: bool | complex,
     xp: Namespace,
     *,
     dtype: Optional[DType] = None,
@@ -80,7 +80,7 @@ def full(
 def full_like(
     x: Array,
     /,
-    fill_value: Union[int, float],
+    fill_value: bool | complex,
     *,
     xp: Namespace,
     dtype: Optional[DType] = None,

array_api_compat/torch/_aliases.py

@@ -602,7 +602,7 @@ def linspace(start: Union[int, float],
 # torch.full does not accept an int size
 # https://github.com/pytorch/pytorch/issues/70906
 def full(shape: Union[int, Tuple[int, ...]],
-         fill_value: Union[bool, int, float, complex],
+         fill_value: bool | complex,
          *,
          dtype: Optional[DType] = None,
          device: Optional[Device] = None,