Fix bugs in LFNA

exps/LFNA/basic-prev.py

@@ -149,10 +149,7 @@ if __name__ == "__main__":
         help="The synthetic enviornment version.",
     )
     parser.add_argument(
-        "--hidden_dim",
-        type=int,
-        required=True,
-        help="The hidden dimension.",
+        "--hidden_dim", type=int, required=True, help="The hidden dimension.",
     )
     parser.add_argument(
         "--init_lr",
@@ -167,16 +164,10 @@ if __name__ == "__main__":
         help="The gap between prev_time and current_timestamp",
     )
     parser.add_argument(
-        "--batch_size",
-        type=int,
-        default=512,
-        help="The batch size",
+        "--batch_size", type=int, default=512, help="The batch size",
     )
     parser.add_argument(
-        "--epochs",
-        type=int,
-        default=300,
-        help="The total number of epochs.",
+        "--epochs", type=int, default=300, help="The total number of epochs.",
     )
     parser.add_argument(
         "--workers",