update hint

Signed-off-by: YunLiu <[email protected]>
Project-MONAI · Sep 30, 2024 · d019799 · d019799
1 parent 195b75a
commit d019799
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/generation/maisi/scripts/diff_model_setting.py b/generation/maisi/scripts/diff_model_setting.py
@@ -74,7 +74,7 @@ def load_config(env_config_path: str, model_config_path: str, model_def_path: st
     return args
 
 
-def initialize_distributed(num_gpus) -> tuple:
+def initialize_distributed(num_gpus: int) -> tuple:
     """
     Initialize distributed training.