@@ -37,9 +37,10 @@ def test_prepare_for_classification_rerun(
3737 self , mock_boto_client , mock_stack_info_class
3838 ):
3939 """Test document preparation for classification rerun"""
40- from idp_cli .rerun_processor import RerunProcessor
4140 from idp_common .models import Document , Page , Section , Status
4241
42+ from idp_cli .rerun_processor import RerunProcessor
43+
4344 mock_stack_info = MagicMock ()
4445 mock_stack_info .validate_stack .return_value = True
4546 mock_stack_info .get_resources .return_value = {
@@ -91,9 +92,10 @@ def test_prepare_for_extraction_rerun(
9192 self , mock_boto_client , mock_stack_info_class
9293 ):
9394 """Test document preparation for extraction rerun"""
94- from idp_cli .rerun_processor import RerunProcessor
9595 from idp_common .models import Document , Page , Section , Status
9696
97+ from idp_cli .rerun_processor import RerunProcessor
98+
9799 mock_stack_info = MagicMock ()
98100 mock_stack_info .validate_stack .return_value = True
99101 mock_stack_info .get_resources .return_value = {
@@ -145,9 +147,10 @@ def test_prepare_for_extraction_rerun(
145147 @patch ("boto3.client" )
146148 def test_send_to_queue (self , mock_boto_client , mock_stack_info_class ):
147149 """Test sending document to SQS queue"""
148- from idp_cli .rerun_processor import RerunProcessor
149150 from idp_common .models import Document , Status
150151
152+ from idp_cli .rerun_processor import RerunProcessor
153+
151154 mock_stack_info = MagicMock ()
152155 mock_stack_info .validate_stack .return_value = True
153156 mock_stack_info .get_resources .return_value = {
0 commit comments