sghorbal commited on
Commit
bac07b6
·
1 Parent(s): 9e7b4b9

fixup: wrong metadata key

Browse files
Files changed (1) hide show
  1. src/main.py +1 -1
src/main.py CHANGED
@@ -135,7 +135,7 @@ async def process_transaction(
135
  db=db,
136
  transaction=transaction,
137
  fraud_score=fraud_output.fraud_probability,
138
- model_version=fraud_output.model_metadata['model_version'] if 'model_version' in fraud_output.model_metadata else 'unknown'
139
  )
140
 
141
  # Send notification to the user
 
135
  db=db,
136
  transaction=transaction,
137
  fraud_score=fraud_output.fraud_probability,
138
+ model_version=fraud_output.model_metadata['version'] if 'version' in fraud_output.model_metadata else 'unknown'
139
  )
140
 
141
  # Send notification to the user