dm-haiku version

[raemisch]

This is more FYI than actual issue.

python 3.9.13
colabdesign 1.1.1

pip will install dm-haiku v0.0.14

With this combination, the colabfold import fails.

Downgrading to dm-haiku==0.0.13 fixes the issue

Error:

  File "venv/lib/python3.9/site-packages/colabdesign/af/alphafold/model/utils.py", line 23, in <module>
    import haiku as hk
  File "venv/lib/python3.9/site-packages/haiku/__init__.py", line 18, in <module>
    from haiku import config
  File "venv/lib/python3.9/site-packages/haiku/config.py", line 18, in <module>
    from haiku._src.config import context
  File "venv/lib/python3.9/site-packages/haiku/_src/config.py", line 50, in <module>
    check_jax_usage: bool | None = None,
TypeError: unsupported operand type(s) for |: 'type' and 'NoneType'