10 lines
373 B
Python
10 lines
373 B
Python
from .utils import get_optimizer, register_optimizer
|
|
|
|
# DMuon is optional (external package). Skip registration if dmuon isn't
|
|
# installed so wall_x still imports; the registry will simply not have
|
|
# "dmuon" and get_optimizer("dmuon", ...) will raise a clear KeyError.
|
|
try:
|
|
from . import dmuon # noqa: F401 - side-effect registers "dmuon"
|
|
except ImportError:
|
|
pass
|