[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

generation/maisi/scripts/diff_model_create_training_data.py

@@ -160,7 +160,9 @@ def process_file(
 
 
 @torch.inference_mode()
-def diff_model_create_training_data(env_config_path: str, model_config_path: str, model_def_path: str, num_gpus: int) -> None:
+def diff_model_create_training_data(
+    env_config_path: str, model_config_path: str, model_def_path: str, num_gpus: int
+) -> None:
     """
     Create training data for the diffusion model.
 
@@ -224,9 +226,7 @@ def diff_model_create_training_data(env_config_path: str, model_config_path: str
     parser.add_argument(
         "--model_def", type=str, default="./configs/config_maisi.json", help="Path to model definition file"
     )
-    parser.add_argument(
-        "--num_gpus", type=int, default=1, help="Number of GPUs to use for distributed training"
-    )
+    parser.add_argument("--num_gpus", type=int, default=1, help="Number of GPUs to use for distributed training")
 
     args = parser.parse_args()
     diff_model_create_training_data(args.env_config, args.model_config, args.model_def, args.num_gpus)

generation/maisi/scripts/diff_model_train.py

@@ -430,9 +430,7 @@ def diff_model_train(env_config_path: str, model_config_path: str, model_def_pat
     parser.add_argument(
         "--model_def", type=str, default="./configs/config_maisi.json", help="Path to model definition file"
     )
-    parser.add_argument(
-        "--num_gpus", type=int, default=1, help="Number of GPUs to use for training"
-    )
+    parser.add_argument("--num_gpus", type=int, default=1, help="Number of GPUs to use for training")
 
     args = parser.parse_args()
     diff_model_train(args.env_config, args.model_config, args.model_def, args.num_gpus)