Fix unit test

Author: pintaoz

test/unit_tests/cli/test_cluster_stack.py

@@ -330,7 +330,7 @@ def test_create_cluster_stack_success(self, mock_hp_cluster_stack_class, mock_lo
 
         from sagemaker.hyperpod.cli.commands.cluster_stack import create_cluster_stack
 
-        create_cluster_stack.callback('config.yaml', 'us-west-2', False)
+        create_cluster_stack.callback('config.yaml', 'us-west-2', 1, False)
 
         mock_load_config.assert_called_once()
         mock_filter.assert_called_once_with({'key': 'value'})
@@ -347,7 +347,7 @@ def test_create_cluster_stack_file_not_found(self, mock_exists, mock_get_templat
 
         from sagemaker.hyperpod.cli.commands.cluster_stack import create_cluster_stack
 
-        create_cluster_stack.callback('nonexistent.yaml', 'us-west-2', False)
+        create_cluster_stack.callback('nonexistent.yaml', 'us-west-2', 1, False)
 
         # Assert - function should return early without error
         mock_exists.assert_called_once_with('nonexistent.yaml')