From 97b107b2cf124f330a06818664de9658a8d3edc1 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 30 Sep 2024 06:46:36 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .../maisi/scripts/diff_model_create_training_data.py | 8 ++++---- generation/maisi/scripts/diff_model_train.py | 4 +--- 2 files changed, 5 insertions(+), 7 deletions(-) diff --git a/generation/maisi/scripts/diff_model_create_training_data.py b/generation/maisi/scripts/diff_model_create_training_data.py index da65d84ba..ca44b43cc 100644 --- a/generation/maisi/scripts/diff_model_create_training_data.py +++ b/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) diff --git a/generation/maisi/scripts/diff_model_train.py b/generation/maisi/scripts/diff_model_train.py index 9e9c417c6..e6bfcdd7c 100644 --- a/generation/maisi/scripts/diff_model_train.py +++ b/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)