diff --git a/ansible_mitogen/strategy.py b/ansible_mitogen/strategy.py index 11fc6d7e..078ba921 100644 --- a/ansible_mitogen/strategy.py +++ b/ansible_mitogen/strategy.py @@ -70,14 +70,13 @@ def wrap_action_loader__get(name, *args, **kwargs): return adorned_klass(*args, **kwargs) -def wrap_connection_loader__get(name, play_context, new_stdin): +def wrap_connection_loader__get(name, play_context, new_stdin, **kwargs): """ While the mitogen strategy is active, rewrite connection_loader.get() calls for the 'ssh' and 'local' transports into corresponding requests for the 'mitogen' connection type, passing the original transport name into it as an argument, so that it can emulate the original type. """ - kwargs = {} if name in ('ssh', 'local', 'docker'): kwargs['original_transport'] = name name = 'mitogen'