message: Local device_id value is any valid value at a POD slice, picks any device from devices() array instead of local_devices().
diff: @@ -165,7 +165,7 @@ PYBIND11_MODULE(tpu_client_extension, m) {
  .def("shape", &PyTpuBuffer::on_host_shape)
  .def("device",
  [](PyTpuBuffer* buffer) -> std::shared_ptr<Device> {
  - return buffer->client()->local_devices()[buffer->device_id()];
  + return buffer->client()->devices()[buffer->device_id()];
  })
  .def("platform", &PyTpuBuffer::platform_name)
  .def("is_deleted", [](const PyTpuBuffer& buffer) {
  