Skip to content

refactor(quad): remove diffrax + docs cleanup#94

Merged
chaoming0625 merged 2 commits into
mainfrom
chore/remove-diffrax
May 25, 2026
Merged

refactor(quad): remove diffrax + docs cleanup#94
chaoming0625 merged 2 commits into
mainfrom
chore/remove-diffrax

Conversation

@chaoming0625

@chaoming0625 chaoming0625 commented May 25, 2026

Copy link
Copy Markdown
Collaborator

Summary

Commit 1 — refactor(quad): remove diffrax integrator family

  • Removes the optional diffrax-backed integrator family from braincell.quad — the explicit, implicit, exponential-Euler, and staggered families already in-tree cover every multi-compartment use case, so the heavy optional dependency was carrying its weight only as duplication.
  • Deletes braincell/quad/_diffrax.py and its test module, drops the 12 diffrax_*_step re-exports + __all__ entries from braincell/quad/__init__.py, and removes "diffrax" from the IntegratorEntry.category enumeration docstring in braincell/quad/_registry.py.
  • Drops the quad = ["diffrax"] extra and the two "diffrax" entries (in all / testing dep groups) from pyproject.toml; scrubs the user-facing mentions in README.md, TODO.md, docs/interface-map.md, and section 5.6 of examples/multi_compartment/quad.ipynb.
  • Historical artifacts (changelog.md, dev/legacy/..., dev/superpowers/..., dated .claude/CLAUDE-26.04.*.md snapshots) were left intact by design.

Commit 2 — chore(docs): fix SPHINXPROJ name and drop stale architecture/publication artifacts

  • docs/Makefile: rename SPHINXPROJ from the leftover "brainpy" to "braincell" so the Sphinx build identifies the project correctly.
  • Remove docs/architecture.xml and examples/multi_compartment/vis_outputs/publication.pdf — generated artifacts left over from an earlier branch.

Test plan

  • python -c "import braincell.quad as q; print(len(q.get_registry().names()))" succeeds and registers no diffrax_* entries
  • pytest braincell/quad/ collects and passes
  • grep -rn diffrax braincell/ pyproject.toml docs/ examples/ README.md TODO.md returns no matches
  • pip install -e ".[all]" does not pull diffrax
  • make -C docs html builds with SPHINXPROJ=braincell

The diffrax-backed step functions in braincell.quad pulled in a heavy
optional dependency for adaptive-stepsize solvers that overlapped with
the in-tree explicit, implicit, exponential-Euler, and staggered
families already covering every multi-compartment use case.

- Deleted braincell/quad/_diffrax.py and its test module.
- Dropped the diffrax_*_step re-exports and __all__ entries from
  braincell/quad/__init__.py and the diffrax mention from the module
  docstring.
- Removed "diffrax" from the IntegratorEntry.category enumeration in
  braincell/quad/_registry.py.
- Removed the diffrax extra and the diffrax entries from the all/
  testing dep groups in pyproject.toml.
- Scrubbed diffrax references from README.md, TODO.md,
  docs/interface-map.md, and the section 5.6 catalog block of
  examples/multi_compartment/quad.ipynb.

@sourcery-ai sourcery-ai Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sorry @chaoming0625, you have reached your weekly rate limit of 500000 diff characters.

Please try again later or upgrade to continue using Sourcery

…ion artifacts

- docs/Makefile: rename SPHINXPROJ from the leftover "brainpy" value to
  "braincell" so the Sphinx build identifies the project correctly.
- Remove docs/architecture.xml and
  examples/multi_compartment/vis_outputs/publication.pdf — generated
  artifacts left over from an earlier branch that should not be tracked.
@chaoming0625 chaoming0625 changed the title refactor(quad): remove diffrax integrator family refactor(quad): remove diffrax + docs cleanup May 25, 2026
@chaoming0625 chaoming0625 requested a review from TLDSZ May 25, 2026 06:48
@chaoming0625 chaoming0625 merged commit 784c8d3 into main May 25, 2026
7 checks passed
@chaoming0625 chaoming0625 deleted the chore/remove-diffrax branch May 25, 2026 07:07
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant