Remove duplicate import of Haiku Module

cgarciae / treex

A Pytree Module system for Deep Learning in JAX

https://cgarciae.github.io/treex/

MIT License

215 stars 17 forks source link

Closed ptigwe closed 2 years ago

ptigwe commented 2 years ago

I believe the import of haiku is meant to be the second one within the try block as it is an optional dependency.

PS: Sorry for yet another 1-line PR

cgarciae commented 2 years ago

Hey @ptigwe ! Thanks a lot, this is actually a big deal as its currently causing issues in Elegy.

LGTM!