Skip to content

Releases: google-research/dataclass_array

v1.5.2

19 Mar 15:25
Compare
Choose a tag to compare
  • Drop Python 3.10 support
  • Fix etree.spec_like, jax.ShapeDtypeStruct,... support
  • Changed jax.tree_util keep paths (for better flatten repr)

v1.5.1

31 Aug 15:57
Compare
Choose a tag to compare
  • Fix look_at compatibility with pytorch
  • Add dca.vectorize_method compatibility for tf.nest/tf.data.

v1.5.0

26 Jul 16:04
Compare
Choose a tag to compare
  • Add tf.nest/tf.data compatibility for DataclassArray.
  • Fix import error due to etils

v1.4.2

11 Jul 10:06
Compare
Choose a tag to compare
  • Add dca.concat method in addition to dca.stack.
  • Now require Python 3.9 (drop 3.8 support)

v1.4.1

20 Mar 16:07
Compare
Choose a tag to compare
  • Add torch==2.0.0 support

v1.4.0

13 Mar 16:13
Compare
Choose a tag to compare
  • Add torch support!
  • Add .cpu(), .cuda(), .to() methods to move the dataclass from
    devices when using torch.
  • Breaking: @dataclass(frozen=True) is now automatically applied

v1.3.0

16 Jan 11:02
Compare
Choose a tag to compare
  • Added: Support for static dca.DataclassArray (dataclasses with only
    static fields).

v1.2.1

24 Nov 16:03
Compare
Choose a tag to compare
  • Fixed: Compatibility with edc.dataclass(auto_cast=True) (fix the 'type' object is not subscriptable error)

v1.2.0

17 Oct 13:09
Compare
Choose a tag to compare
  • Changed: By default, dataclass_array do not cast and broadcast inputs
    anymore.
  • Changed: dca.DataclassArray fields can be annotated with named axis (e.g.
    FloatArray['*shape h w 3']). Note that consistency across fields is not
    checked yet.
  • Added: @dca.dataclass_array to customize the dca.DataclassArray params

v1.1.0

24 Aug 11:37
Compare
Choose a tag to compare
  • Added: Array types can be imported directly from dataclass_array.typing
  • Added: Syntax to specify the shape of the DataclassArray (e.g. MyRay['h w']).
  • Fixed: Correctly forward non-init fields in .replace, tree_map,
    @dca.vectorize_method