Skip to content

Commit 0a07aad

Browse files
HAOCHENYEMGAMZ
authored andcommitted
[Lint] Fix lint
1 parent 363492a commit 0a07aad

File tree

2 files changed

+9
-5
lines changed

2 files changed

+9
-5
lines changed

mmengine/optim/optimizer/builder.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def register_sophia_optimizers() -> List[str]:
118118
Returns:
119119
List[str]: A list of registered optimizers' name.
120120
"""
121-
optimizers: List[str] = []
121+
optimizers = [] # type: ignore
122122
try:
123123
import Sophia
124124
except ImportError:
@@ -131,8 +131,7 @@ def register_sophia_optimizers() -> List[str]:
131131
try:
132132
OPTIMIZERS.register_module(module=_optim)
133133
except Exception as e:
134-
warnings.warn(
135-
f"Failed to import {_optim.__name__} for {e}")
134+
warnings.warn(f'Failed to import {Sophia} for {e}')
136135
return optimizers
137136

138137

@@ -165,7 +164,7 @@ def register_bitsandbytes_optimizers() -> List[str]:
165164
try:
166165
OPTIMIZERS.register_module(module=optim_cls, name=name)
167166
except Exception as e:
168-
warnings.warn(f"Failed to import {optim_cls.__name__} for {e}")
167+
warnings.warn(f'Failed to import {optim_cls.__name__} for {e}')
169168
dadaptation_optimizers.append(name)
170169
return dadaptation_optimizers
171170

@@ -183,7 +182,7 @@ def register_transformers_optimizers() -> List[str]:
183182
try:
184183
OPTIMIZERS.register_module(name='Adafactor', module=Adafactor)
185184
except Exception as e:
186-
warnings.warn(f"Failed to import {Adafactor.__name__} for {e}")
185+
warnings.warn(f'Failed to import Adafactor for {e}')
187186
transformer_optimizers.append('Adafactor')
188187
return transformer_optimizers
189188

mmengine/utils/package_utils.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,11 @@ def is_installed(package: str) -> bool:
2222
# For Python 3.7, importlib_metadata backport can be used
2323
import importlib.util
2424

25+
import pkg_resources # type: ignore
26+
27+
# refresh the pkg_resources
28+
# more datails at https://github.com/pypa/setuptools/issues/373
29+
importlib.reload(pkg_resources)
2530
try:
2631
distribution(package)
2732
return True

0 commit comments

Comments
 (0)