remove unnecessary if statement

PiperOrigin-RevId: 617653292

Author: jax authors

jax/experimental/array_serialization/serialization.py

@@ -260,10 +260,7 @@ async def _write_array(shard):
       else:
         await write_future.commit
 
-  if isinstance(arr_inp, array.ArrayImpl):
-    local_shards = arr_inp.addressable_shards
-  else:
-    local_shards = arr_inp.addressable_shards
+  local_shards = arr_inp.addressable_shards
   future_write_state = jax.tree_util.tree_map(_write_array, local_shards)
   return await asyncio.gather(*future_write_state)