From d01979970acd67d4b4ad1cfd0303b20923bc2482 Mon Sep 17 00:00:00 2001
From: YunLiu <55491388+KumoLiu@users.noreply.github.com>
Date: Mon, 30 Sep 2024 15:21:04 +0800
Subject: [PATCH] update hint

Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com>
---
 generation/maisi/scripts/diff_model_setting.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/generation/maisi/scripts/diff_model_setting.py b/generation/maisi/scripts/diff_model_setting.py
index 9ea0bfafb..6ba468886 100644
--- 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.