tensorflow / tensorflow/probability
Can you break your dependency on dm-tree?
Nobody has claimed this yet.
- Dominant language
- Jupyter Notebook
- Stars
- 4.4k
- Forks
- 1.1k
- PR merge metrics
- No merged PRs in 30d
Description
DM-tree doesn't support Python 3.13, which prevents tensorflow-probability from installing on Python 3.13 and therefore all projects that depend on it.
Moreover, DM-tree appears to be unmaintained since they don't even officially support Python 3.12 (based on their pyproject trove classifiers).
Based on the commit log, it appears to be more-or-less unmaintained.
Is there any way to remove this package as a dependency? It appears you only use it one place and a lot of the functionality is already provided by jax.tree.map/flatten, etc. You may also want to look at Equinox, which is actively maintained. Or just implement the functionality in Python since speed is probably not that important, and not worth the maintenance hassle.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Locate the single use of dm-tree in the TensorFlow Probability codebase and compare its required operations with jax.tree.map/flatten and the alternatives mentioned in the issue. Done means removing the dependency while preserving the existing behavior and allowing installation on Python 3.13.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, tensorflow
- Domain
- machine-learning
- Issue type
- Refactor
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100