diff --git a/mev_inspect/crud/classified_traces.py b/mev_inspect/crud/classified_traces.py index 25c6e2b..78f51aa 100644 --- a/mev_inspect/crud/classified_traces.py +++ b/mev_inspect/crud/classified_traces.py @@ -26,24 +26,22 @@ def write_classified_traces( for trace in classified_traces: models.append( ClassifiedTraceModel( - **{ - "transaction_hash": trace.transaction_hash, - "block_number": trace.block_number, - "classification": trace.classification.value, - "trace_type": trace.type.value, - "trace_address": trace.trace_address, - "protocol": str(trace.protocol), - "abi_name": trace.abi_name, - "function_name": trace.function_name, - "function_signature": trace.function_signature, - "inputs": json.loads(trace.json(include={"inputs"}))["inputs"], - "from_address": trace.from_address, - "to_address": trace.to_address, - "gas": trace.gas, - "value": trace.value, - "gas_used": trace.gas_used, - "error": trace.error, - } + transaction_hash=trace.transaction_hash, + block_numbe=trace.block_number, + classification=trace.classification.value, + trace_type=trace.type.value, + trace_address=trace.trace_address, + protocol=str(trace.protocol), + abi_name=trace.abi_name, + function_name=trace.function_name, + function_signature=trace.function_signature, + inputs=json.loads(trace.json(include={"inputs"}))["inputs"], + from_address=trace.from_address, + to_address=trace.to_address, + gas=trace.gas, + value=trace.value, + gas_used=trace.gas_used, + error=trace.error, ) )