search

google / objax

Apache License 2.0
768 stars 77 forks source link

[Jax 0.4.27] AttributeError: arr.device_buffers has been deprecated. #270

Open GaetanLepage opened 4 months ago

GaetanLepage commented 4 months ago

Since jax 0.4.27, several tests fail with:

    @property
    def device_buffers(self):
>     raise AttributeError(
        "arr.device_buffers has been deprecated. Use [x.data for x in arr.addressable_shards]")
E     AttributeError: arr.device_buffers has been deprecated. Use [x.data for x in arr.addressable_shards]