patrick-kidger / sympy2jax

Turn SymPy expressions into trainable JAX expressions.
Apache License 2.0
314 stars 12 forks source link

Add custom concatenate/stack functions #8

Closed calbach closed 1 year ago

calbach commented 1 year ago

This is to support stacking / concatenating arrays within a sympy expression.

For discussion:

patrick-kidger commented 1 year ago

Looks like the pre-commit checks failed.

FYI you can arrange to have these run automatically whenever you commit. They will either pass and commit, or fail and autoformat.

If the latter, you can then check you're happy with the autoformat, followed by git add -u; git commit to add the changes and commit for real.

(See CONTRIBUTING.MD.)

calbach commented 1 year ago

Fixed, I missed this repo was using isort. I'll try out the precommit hook next time.

patrick-kidger commented 1 year ago

Merged! Ty.