File tree Expand file tree Collapse file tree 2 files changed +9
-5
lines changed Expand file tree Collapse file tree 2 files changed +9
-5
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments