diff --git a/src/gluonts/dataset/artificial/_base.py b/src/gluonts/dataset/artificial/_base.py index 1979ad3a01..840407e093 100644 --- a/src/gluonts/dataset/artificial/_base.py +++ b/src/gluonts/dataset/artificial/_base.py @@ -442,8 +442,8 @@ def _get_start(self, index: int, my_random: random.Random) -> str: start_y = my_random.randint(2000, 2018) start_m = my_random.randint(1, 12) start_d = my_random.randint(1, 28) - start_h = my_random.randint(0, 24) - start_min = my_random.randint(0, 60) + start_h = my_random.randint(0, 23) + start_min = my_random.randint(0, 59) else: start_y, start_m, start_d = 2013, 11, 28 start_h, start_min = 18, 36 diff --git a/src/gluonts/distribution/binned.py b/src/gluonts/distribution/binned.py index dfb324894f..7d4ed1b19a 100644 --- a/src/gluonts/distribution/binned.py +++ b/src/gluonts/distribution/binned.py @@ -65,7 +65,7 @@ def _compute_edges(F, bin_centers: Tensor) -> Tensor: Returns ------- Tensor - Tensor of shape (*gamma.shape, num_bins+1) + Tensor of shape (*batch.shape, num_bins+1) """ low = (