HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Code refactor https://ultralytics.com/actions (#13844)

Browse source
This commit is contained in:
Glenn Jocher 2024-06-21 01:10:52 +02:00 • committed by GitHub
parent ffb46fd7fb
commit ecfcdf12c9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
13 changed files with 21 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

6
ultralytics/nn/tasks.py
View file

@ -693,7 +693,7 @@ class Ensemble(nn.ModuleList):
@contextlib.contextmanager
def temporary_modules(modules={}, attributes={}):
def temporary_modules(modules=None, attributes=None):
"""
Context manager for temporarily adding or modifying modules in Python's module cache (`sys.modules`).
@ -718,6 +718,10 @@ def temporary_modules(modules={}, attributes={}):
applications or libraries. Use this function with caution.
"""
if modules is None:
modules = {}
if attributes is None:
attributes = {}
import sys
from importlib import import_module