[JAX] Fail gracefully when an array with multiple shards is passed to make_array_from_single_device_arrays.
[JAX] Fail gracefully when an array with multiple shards is passed to make_array_from_single_device_arrays.

Fix a crash when an exception is thrown during PyArray construction.
