ImportError: cannot import name 'MuonWithAuxAdam' from 'muon'

[shiwanghua]

(grpo) ➜  grpo_exe pip install git+https://github.com/KellerJordan/Muon
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Collecting git+https://github.com/KellerJordan/Muon
  Cloning https://github.com/KellerJordan/Muon to /tmp/pip-req-build-ycfs1fph
  Running command git clone --filter=blob:none --quiet https://github.com/KellerJordan/Muon /tmp/pip-req-build-ycfs1fph
  Resolved https://github.com/KellerJordan/Muon to commit 857d1adc159e098212f828d927460782d3da59e9
  Preparing metadata (setup.py) ... done

[notice] A new release of pip is available: 25.2 -> 25.3
[notice] To update, run: pip install --upgrade pip

(grpo) ➜  grpo_exe python3
Python 3.11.9 (main, Apr 19 2024, 16:48:06) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
Ctrl click to launch VS Code Native REPL
>>> from muon import MuonWithAuxAdam
/home/ubisec/miniconda3/envs/grpo/lib/python3.11/site-packages/muon/_core/preproc.py:31: FutureWarning: `__version__` is deprecated, use `importlib.metadata.version('scanpy')` instead
  if Version(scanpy.__version__) < Version("1.10"):
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ImportError: cannot import name 'MuonWithAuxAdam' from 'muon' (/home/ubisec/miniconda3/envs/grpo/lib/python3.11/site-packages/muon/__init__.py)
>>>