add default values to lower/upper bounds

awslabs · Aug 22, 2023 · 0e65431 · 0e65431
1 parent e287242
commit 0e65431
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/gluonts/torch/distributions/truncated_normal.py b/src/gluonts/torch/distributions/truncated_normal.py
@@ -86,8 +86,8 @@ def __init__(
  self,
  loc: torch.Tensor,
  scale: torch.Tensor,
- min: Union[torch.Tensor, float],
- max: Union[torch.Tensor, float],
+ min: Union[torch.Tensor, float] = -1.0,
+ max: Union[torch.Tensor, float] = 1.0,
  upscale: Union[torch.Tensor, float] = 5.0,
  tanh_loc: bool = False,
  ):
@@ -240,8 +240,8 @@ class TruncatedNormalOutput(DistributionOutput):
  @validated()
  def __init__(
  self,
- min: float,
- max: float,
+ min: float = -1.0,
+ max: float = 1.0,
  upscale: float = 5.0,
  tanh_loc: bool = False,
  ) -> None: