WIP

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 = (