8000 Jax v0.6.0 Causes Unexpected Keyword Argument Error · Issue #116 · wilson-labs/cola · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

Jax v0.6.0 Causes Unexpected Keyword Argument Error #116

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Open
adam-hartshorne opened this issue Apr 24, 2025 · 0 comments
Open

Jax v0.6.0 Causes Unexpected Keyword Argument Error #116

adam-hartshorne opened this issue Apr 24, 2025 · 0 comments
Labels
bug Something isn't working

Comments

@adam-hartshorne
Copy link

🐛 Bug

simply calling "import cola" causes a unexpected keyword argument error with the new Jax v0.6.0

To reproduce

** Code snippet to reproduce **

# Your code goes here
# Please make sure it does not require any external dependencies (other than JaX/PyTorch!)
# (We much prefer small snippets rather than links to existing libraries!)

** Stack trace/error message **

    import cola
  File "/home/adam/anaconda3/envs/jax_latest/lib/python3.12/site-packages/cola/__init__.py", line 11, in <module>
    import_from_all("fns", globals(), __all__, __name__)
  File "/home/adam/anaconda3/envs/jax_latest/lib/python3.12/site-packages/cola/utils/__init__.py", line 37, in import_from_all
    module = importlib.import_module('.' + module_name, package=_name)
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/adam/anaconda3/envs/jax_latest/lib/python3.12/importlib/__init__.py", line 90, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/adam/anaconda3/envs/jax_latest/lib/python3.12/site-packages/cola/fns.py", line 155, in <module>
    @dispatch(cond=lambda A: A.isa(cola.SelfAdjoint))
     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: Dispatcher.__call__() got an unexpected keyword argument 'cond'

System information

Please complete the following information:
Cola v0.0.7
JAX v0.6.0
Python 3.12
Cuda 12.8
@adam-hartshorne adam-hartshorne added the bug Something isn't working label Apr 24, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
bug Something isn't working
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@adam-hartshorne
0