Fix black issues

exps/LFNA/lfna.py

@@ -101,7 +101,10 @@ def main(args):
     )
     lr_scheduler = torch.optim.lr_scheduler.MultiStepLR(
         optimizer,
-        milestones=[int(args.epochs * 0.8), int(args.epochs * 0.9),],
+        milestones=[
+            int(args.epochs * 0.8),
+            int(args.epochs * 0.9),
+        ],
         gamma=0.1,
     )
     logger.log("The base-model is\n{:}".format(base_model))
@@ -240,13 +243,22 @@ if __name__ == "__main__":
         help="The synthetic enviornment version.",
     )
     parser.add_argument(
-        "--hidden_dim", type=int, default=16, help="The hidden dimension.",
+        "--hidden_dim",
+        type=int,
+        default=16,
+        help="The hidden dimension.",
     )
     parser.add_argument(
-        "--layer_dim", type=int, default=16, help="The layer chunk dimension.",
+        "--layer_dim",
+        type=int,
+        default=16,
+        help="The layer chunk dimension.",
     )
     parser.add_argument(
-        "--time_dim", type=int, default=16, help="The timestamp dimension.",
+        "--time_dim",
+        type=int,
+        default=16,
+        help="The timestamp dimension.",
     )
     #####
     parser.add_argument(
@@ -262,7 +274,10 @@ if __name__ == "__main__":
         help="The weight decay for the optimizer (default is Adam)",
     )
     parser.add_argument(
-        "--meta_batch", type=int, default=64, help="The batch size for the meta-model",
+        "--meta_batch",
+        type=int,
+        default=64,
+        help="The batch size for the meta-model",
     )
     parser.add_argument(
         "--sampler_enlarge",
@@ -284,7 +299,10 @@ if __name__ == "__main__":
         "--workers", type=int, default=4, help="The number of workers in parallel."
     )
     parser.add_argument(
-        "--device", type=str, default="cpu", help="",
+        "--device",
+        type=str,
+        default="cpu",
+        help="",
     )
     # Random Seed
     parser.add_argument("--rand_seed", type=int, default=-1, help="manual seed")