diff options
| author | iamsoto <theintrocode@gmail.com> | 2021-01-08 18:04:11 -0800 |
|---|---|---|
| committer | iamsoto <theintrocode@gmail.com> | 2021-01-08 18:04:11 -0800 |
| commit | aa6f5b76e60e7ba71b278e9c0125c35e5f8a92e7 (patch) | |
| tree | af285e41cf253755133292654a39f594b0a4689e | |
| parent | 73fe877ff967f279d470b81ad447b9f3056c1335 (diff) | |
| download | numpy-aa6f5b76e60e7ba71b278e9c0125c35e5f8a92e7.tar.gz | |
ENH: keyboard interrupt for large array creation
| -rw-r--r-- | numpy/core/src/multiarray/array_coercion.c | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/numpy/core/src/multiarray/array_coercion.c b/numpy/core/src/multiarray/array_coercion.c index 1eac401bc..0fd958282 100644 --- a/numpy/core/src/multiarray/array_coercion.c +++ b/numpy/core/src/multiarray/array_coercion.c @@ -1098,6 +1098,11 @@ PyArray_DiscoverDTypeAndShape_Recursive( return curr_dims + 1; } + /* Allow keyboard interrupts. See gh issue 18117. */ + if (PyErr_CheckSignals() < 0) { + return -1; + } + /* Recursive call for each sequence item */ for (Py_ssize_t i = 0; i < size; i++) { max_dims = PyArray_DiscoverDTypeAndShape_Recursive( |