[Typing][B-26,B-27] Add type annotations for `python/paddle/distribut…

…ion/{transform,transformed_distribution}.py` (PaddlePaddle#65912) --------- Co-authored-by: SigureMo <sigure.qaq@gmail.com>
lixcli · Jul 22, 2024 · 869858e · 869858e
1 parent b022ab4
commit 869858e
Show file tree

Hide file tree

Showing 3 changed files with 160 additions and 116 deletions.
diff --git a/python/paddle/distribution/constraint.py b/python/paddle/distribution/constraint.py
@@ -34,7 +34,7 @@ def __call__(self, value: Tensor) -> Tensor:
 
 
 class Range(Constraint):
-    def __init__(self, lower: Tensor, upper: Tensor) -> None:
+    def __init__(self, lower: float | Tensor, upper: float | Tensor) -> None:
         self._lower = lower
         self._upper = upper
         super().__init__()