Fix broken tests for equivalent_argparse_code

Signed-off-by: Fabrice Normandin <[email protected]>

Author: lebrice

examples/simple/basic.py

@@ -49,7 +49,10 @@ class HParams:
 expected += """
 parser = ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
 
-group = parser.add_argument_group(title="HParams ['hparams']", description="Set of options for the training of a Model.")
+group = parser.add_argument_group(
+    title="HParams ['hparams']",
+    description="Set of options for the training of a Model.",
+)
 group.add_argument(*['--num_layers'], **{'type': int, 'required': False, 'dest': 'hparams.num_layers', 'default': 4, 'help': ' '})
 group.add_argument(*['--num_units'], **{'type': int, 'required': False, 'dest': 'hparams.num_units', 'default': 64, 'help': ' '})
 group.add_argument(*['--optimizer'], **{'type': str, 'required': False, 'dest': 'hparams.optimizer', 'default': 'ADAM', 'help': ' '})

examples/simple/inheritance.py

@@ -58,7 +58,10 @@ class MAML(Method):
 expected += """
 parser = ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
 
-group = parser.add_argument_group(title="MAML ['hparams']", description="Overwrites some of the default values and adds new arguments/attributes.")
+group = parser.add_argument_group(
+    title="MAML ['hparams']",
+    description="Overwrites some of the default values and adds new arguments/attributes.",
+)
 group.add_argument(*['--num_layers'], **{'type': int, 'required': False, 'dest': 'hparams.num_layers', 'default': 6, 'help': ' '})
 group.add_argument(*['--num_units'], **{'type': int, 'required': False, 'dest': 'hparams.num_units', 'default': 128, 'help': ' '})
 group.add_argument(*['--optimizer'], **{'type': str, 'required': False, 'dest': 'hparams.optimizer', 'default': 'ADAM', 'help': ' '})