-parser = argparse.ArgumentParser(description='Toy attention model.')
-
-parser.add_argument('--nb_epochs',
- type = int, default = 250)
-
-parser.add_argument('--with_attention',
- help = 'Use the model with an attention layer',
- action='store_true', default=False)
-
-parser.add_argument('--group_by_locations',
- help = 'Use the task where the grouping is location-based',
- action='store_true', default=False)
-
-parser.add_argument('--positional_encoding',
- help = 'Provide a positional encoding',
- action='store_true', default=False)
-
-parser.add_argument('--seed',
- type = int, default = 0,
- help = 'Random seed (default 0, < 0 is no seeding)')
+parser = argparse.ArgumentParser(description="Toy attention model.")
+
+parser.add_argument("--nb_epochs", type=int, default=250)
+
+parser.add_argument(
+ "--with_attention",
+ help="Use the model with an attention layer",
+ action="store_true",
+ default=False,
+)
+
+parser.add_argument(
+ "--group_by_locations",
+ help="Use the task where the grouping is location-based",
+ action="store_true",
+ default=False,
+)
+
+parser.add_argument(
+ "--positional_encoding",
+ help="Provide a positional encoding",
+ action="store_true",
+ default=False,
+)
+
+parser.add_argument(
+ "--seed", type=int, default=0, help="Random seed (default 0, < 0 is no seeding)"
+)