From 6620770665f8aaf7abaaeb85de7deca4515bc619 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Tue, 18 Feb 2020 14:58:23 -0800 Subject: [PATCH] Simpler approach to loading entrypoints in hubconf works properly --- hubconf.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/hubconf.py b/hubconf.py index e58b2cb4..70fed79a 100644 --- a/hubconf.py +++ b/hubconf.py @@ -1,10 +1,4 @@ dependencies = ['torch'] - from timm.models import registry -current_module = __import__(__name__) -current_module.__dict__.update(registry._model_entrypoints) -#for fn_name in registry.list_models(): -# fn = registry.model_entrypoint(fn_name) -# setattr(current_module, fn_name, fn) - +globals().update(registry._model_entrypoints)