Changed TypeError wording per PR feedback

IntelPython · Nov 7, 2023 · da59476 · da59476
1 parent aadb6b4
commit da59476
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/dpctl/tensor/_usmarray.pyx b/dpctl/tensor/_usmarray.pyx
@@ -191,10 +191,11 @@ cdef class usm_ndarray:
                 try:
                     <Py_ssize_t> shape
                     shape = [shape, ]
-                except Exception:
+                except Exception as e:
                     raise TypeError(
-                        "Argument shape must be a list or a tuple."
-                    )
+                        "Argument shape must a non-negative integer, "
+			"or a list/tuple of such integers."
+                    ) from e
         nd = len(shape)
         if dtype is None:
             if isinstance(buffer, (dpmem._memory._Memory, usm_ndarray)):