Fix legacy adaptor .integers()

Project-MONAI · Aug 23, 2023 · c564ded · c564ded
1 parent 3372f89
commit c564ded
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/monai/utils/utils_random_generator_adaptor.py b/monai/utils/utils_random_generator_adaptor.py
@@ -73,7 +73,12 @@ def __init__(self, seed: int | None = None, random_state: np.random.RandomState
     def integers(
         self, low: int, high: int | None = None, size: ShapeLike | None = None, dtype=DtypeLike, endpoint: bool = False
     ) -> NdarrayTensor:
-        return self.random_state.randint(low=low, high=high if endpoint else high + 1, size=size, dtype=dtype)
+        if endpoint:
+            if high is not None:
+                high += 1
+            else:
+                low += 1
+        return self.random_state.randint(low=low, high=high, size=size, dtype=dtype)
 
     def random(self, size: ShapeLike | None = None, dtype=DtypeLike, out: NdarrayTensor | None = None) -> NdarrayTensor:
         if out is not None: