Skip to content

Commit 18aa00a

Browse files
author
Bob Strahan
committed
Merge branch 'develop' of ssh.gitlab.aws.dev:genaiic-reusable-assets/engagement-artifacts/genaiic-idp-accelerator into develop
2 parents 4e48aef + 2a23ea3 commit 18aa00a

File tree

5 files changed

+13
-12
lines changed

5 files changed

+13
-12
lines changed

idp_cli/tests/test_batch_processor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from unittest.mock import MagicMock, patch
1010

1111
import pytest
12+
1213
from idp_cli.batch_processor import BatchProcessor
1314

1415

idp_cli/tests/test_manifest_parser.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
import json
1010

1111
import pytest
12+
1213
from idp_cli.manifest_parser import ManifestParser, parse_manifest, validate_manifest
1314

1415

idp_cli/tests/test_progress_monitor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
from unittest.mock import MagicMock, patch
1010

1111
import pytest
12+
1213
from idp_cli.progress_monitor import ProgressMonitor
1314

1415

idp_cli/tests/test_rerun_processor.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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 = {

template.yaml

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -929,16 +929,11 @@ Resources:
929929
AppSyncApiArn: !GetAtt GraphQLApi.Arn
930930
ConfigurationTable: !Ref ConfigurationTable
931931
ConfigurationDefaultS3Uri: !If
932-
- HasCustomConfigPath
933-
- !Ref CustomConfigPath
934-
- !Sub
932+
- HasPattern2CustomConfigPath
933+
- !Sub "s3://${ConfigurationBucket}/${Pattern2CustomConfigPath}/config.yaml"
934+
- !Sub
935935
- "s3://${ConfigurationBucket}/config_library/pattern-2/${ConfigPath}/config.yaml"
936-
- ConfigPath:
937-
!FindInMap [
938-
Pattern2ConfigurationMap,
939-
!Ref Pattern2Configuration,
940-
ConfigPath,
941-
]
936+
- ConfigPath: !FindInMap [Pattern2ConfigurationMap, !Ref Pattern2Configuration, ConfigPath]
942937
ConfigLibraryHash: "<CONFIG_LIBRARY_HASH_TOKEN>"
943938
EnableHITL: !Ref EnableHITL
944939
SageMakerA2IReviewPortalURL: !If

0 commit comments

Comments
 (0)