From 80ac53e72e47a4fe1414791274ac8d0c9fbb0bdd Mon Sep 17 00:00:00 2001 From: Ian Adam Naval Date: Fri, 20 Feb 2015 22:54:09 -0500 Subject: [PATCH] Fix example commands --- example_cmd.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/example_cmd.py b/example_cmd.py index 35d3df2..75385a8 100644 --- a/example_cmd.py +++ b/example_cmd.py @@ -4,17 +4,17 @@ from commands import BaseCommand, register_cmd class example_cmd(BaseCommand): def __call__(self, *args, **kwargs): - def input_generator(): - yield 'example' - yield 'command' - return input_generator + def output_generator(): + yield b'example' + yield b'command' + return output_generator() @register_cmd class echo(BaseCommand): def __call__(self, *args, **kwargs): - def input_generator(): + def output_generator(): for line in self.input_cmd(): yield line - return input_generator + return output_generator()