diff --git a/ansible_mitogen/strategy.py b/ansible_mitogen/strategy.py index a9179f737..0c92b9bc8 100644 --- a/ansible_mitogen/strategy.py +++ b/ansible_mitogen/strategy.py @@ -45,12 +45,11 @@ import ansible_mitogen.process import ansible.executor.process.worker +import ansible.plugins.loader import ansible.template import ansible.utils.sentinel import ansible.playbook.play_context -from ansible.plugins.loader import get_with_context_result - def _patch_awx_callback(): """ @@ -103,10 +102,10 @@ def wrap_action_loader__get_with_context(name, *args, **kwargs): bases = (ansible_mitogen.mixins.ActionModuleMixin, klass) adorned_klass = type(str(name), bases, {}) if kwargs.get('class_only'): - return get_with_context_result(adorned_klass, context) - return get_with_context_result(adorned_klass(*args, **kwargs), context) + return ansible.plugins.loader.get_with_context_result(adorned_klass, context) + return ansible.plugins.loader.get_with_context_result(adorned_klass(*args, **kwargs), context) - return get_with_context_result(None, context) + return ansible.plugins.loader.get_with_context_result(None, context) REDIRECTED_CONNECTION_PLUGINS = (