tannedcung/sbt-idp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

force predicted values to None or empty base on doc type

Browse Source
This commit is contained in:
PhanThanhTrung 2024-03-11 16:42:51 +07:00
parent b8e37068f6
commit c3d177dc0c
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
cope2n-api/fwd_api/celery_worker/process_result_tasks.py
View File

@ -210,13 +210,20 @@ def _update_subscription_rq_file(request_id, index_in_request, doc_type, result)
purchase_date = elem['value'] purchase_date = elem['value']
else: else:
imei_number = elem['value'] imei_number = elem['value']
if doc_type=='invoice':
_predict_result = { _predict_result = {
"retailername": retailer_name, "retailername": retailer_name,
"sold_to_party": sold_to_party, "sold_to_party": sold_to_party,
"purchase_date": purchase_date, "purchase_date": purchase_date,
"imei_number": imei_number "imei_number": []
} }
else:
_predict_result = {
"retailername": None,
"sold_to_party": None,
"purchase_date": [],
"imei_number": imei_number
}
image.predict_result = _predict_result image.predict_result = _predict_result
image.save() image.save()