actually use auto_import_modules correctly

2021-01-01 02:59:42 +01:00
parent c64e33173a
commit 92d96ac336
2 changed files with 4 additions and 5 deletions
@@ -120,7 +120,7 @@ def import_module_decorator(func):
                )
            inference_state.module_cache.add(import_names, python_value_set)

-        if not prefer_stubs:
+        if not prefer_stubs or import_names[0] in settings.auto_import_modules:
            return python_value_set

        stub = try_to_load_stub_cached(inference_state, import_names, python_value_set,