diff --git a/.github/workflows/deploy.yml b/.github/workflows/deploy.yml index 3b1443b..978e4d1 100755 --- a/.github/workflows/deploy.yml +++ b/.github/workflows/deploy.yml @@ -7,7 +7,7 @@ on: env: AWS_REGION: ap-northeast-2 ECR_REPOSITORY: imfact-inferer - IMAGE_TAG: latest + IMAGE_TAG: dev AWS_ACCOUNT_ID: ${{ secrets.AWS_ACCOUNT_ID }} jobs: diff --git a/imfact_batch_infer.py b/imfact_batch_infer.py index f8e24e3..d2f16a1 100644 --- a/imfact_batch_infer.py +++ b/imfact_batch_infer.py @@ -5,6 +5,7 @@ from concurrent.futures import ThreadPoolExecutor import boto3 from tqdm import tqdm +from typing import Any, Dict from imfact_infer import ImFactInferer @@ -29,7 +30,7 @@ def init_inferer(): return _inferer -def process_single_article(article: dict): +def process_single_article(article: Dict[str, Any]) -> Dict[str, Any]: inferer = init_inferer() result = inferer.compute_news_reliability(article["title"], article["body"]) article["reliability_score"] = result["final_score"]