Fix exception context in BoundedArray.__init__

google-deepmind · Mar 11, 2021 · 7478092 · 7478092
1 parent 09686eb
commit 7478092
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/dm_env/specs.py b/dm_env/specs.py
@@ -207,14 +207,14 @@ def __init__(self, shape, dtype, minimum, maximum, name=None):
     try:
       bcast_minimum = np.broadcast_to(minimum, shape=shape)
     except ValueError as numpy_exception:
-      raise ValueError('minimum is not compatible with shape. '
-                       'Message: {!r}.'.format(numpy_exception))
+      raise ValueError('minimum is not compatible with shape. Message: '
+                       '{!r}.'.format(numpy_exception)) from numpy_exception
 
     try:
       bcast_maximum = np.broadcast_to(maximum, shape=shape)
     except ValueError as numpy_exception:
-      raise ValueError('maximum is not compatible with shape. '
-                       'Message: {!r}.'.format(numpy_exception))
+      raise ValueError('maximum is not compatible with shape. Message: '
+                       '{!r}.'.format(numpy_exception)) from numpy_exception
 
     if np.any(bcast_minimum > bcast_maximum):
       raise ValueError(_MINIMUM_MUST_BE_LESS_THAN_OR_EQUAL_TO_MAXIMUM.format(