File tree Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Original file line number Diff line number Diff line change @@ -129,14 +129,15 @@ def _recompile(func=None, fastmath=None):
129
129
None
130
130
"""
131
131
warnings .warn (CACHE_WARNING )
132
- njit_funcs = get_njit_funcs ()
132
+
133
+ njit_funcs = []
134
+ for module_name , func_name in get_njit_funcs ():
135
+ module = importlib .import_module (f".{ module_name } " , package = "stumpy" )
136
+ njit_funcs .append (getattr (module , func_name ))
133
137
134
138
recompile_funcs = []
135
139
if func is None :
136
- for module_name , func_name in njit_funcs :
137
- module = importlib .import_module (f".{ module_name } " , package = "stumpy" )
138
- func = getattr (module , func_name )
139
- recompile_funcs .append (func )
140
+ recompile_funcs = njit_funcs
140
141
elif func in njit_funcs :
141
142
recompile_funcs .append (func )
142
143
else :
You can’t perform that action at this time.
0 commit comments