Skip to content

Commit

Permalink
revert __format__
Browse files Browse the repository at this point in the history
  • Loading branch information
ASEM000 committed Mar 19, 2024
1 parent b67ed03 commit 3f92f91
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -33,12 +33,6 @@
- `freeze` -> Use `ft.partial(tree_mask, lambda _: True)` instead.
- `unfreeze` -> Use `tree_unmask` instead.

- Implement `__format__` for `TreeClass` for shorter syntax representation:
- `f"{tree:3r}"` is tree **r**epr of depth = 3
- `f"{tree:3s}"` is tree **s**tr of depth = 3
- `f"{tree:3d}"` is tree **d**iagram of depth = 3
- `f"{tree:3t}`" is tree summary of depth = 3

- `tree_{mask,unmask}` now accepts only callable `cond` argument.
- Rename `is_frozen` to `is_masked`
- frozen could mean non-trainable array, however the masking is not only for arrays but also for other types that will be hidden from jax transformations.
Expand Down

0 comments on commit 3f92f91

Please sign in to comment.