[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>
inaomIIsfarell · Jul 31, 2024 · 3f345de · 3f345de
1 parent 81bf297
commit 3f345de
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__()