diff --git a/mev_inspect/classifiers/trace.py b/mev_inspect/classifiers/trace.py index df2fdb5..2467346 100644 --- a/mev_inspect/classifiers/trace.py +++ b/mev_inspect/classifiers/trace.py @@ -6,8 +6,8 @@ from mev_inspect.schemas.blocks import CallAction, CallResult, Trace, TraceType from mev_inspect.schemas.classified_traces import ( Classification, ClassifiedTrace, - Call, - ClassifiedCall, + CallTrace, + ClassifiedCallTrace, ClassifierSpec, ) @@ -71,7 +71,7 @@ class TraceClassifier: signature, Classification.unknown ) - return ClassifiedCall( + return ClassifiedCallTrace( **trace.dict(), trace_type=trace.type, classification=classification, @@ -87,7 +87,7 @@ class TraceClassifier: gas_used=result.gas_used if result is not None else None, ) - return Call( + return CallTrace( **trace.dict(), trace_type=trace.type, classification=Classification.unknown, diff --git a/mev_inspect/schemas/classified_traces.py b/mev_inspect/schemas/classified_traces.py index 29cbf6e..57a5500 100644 --- a/mev_inspect/schemas/classified_traces.py +++ b/mev_inspect/schemas/classified_traces.py @@ -28,7 +28,7 @@ class Protocol(Enum): curve = "curve" -# -------- Trace Models ------------------------------------------------------------------------------ +# -------- Traces ------------------------------------------------------------------------------ class ClassifiedTrace(Trace): @@ -52,13 +52,13 @@ class ClassifiedTrace(Trace): abi_name: Optional[str] -class Call(ClassifiedTrace): +class CallTrace(ClassifiedTrace): to_address: str from_address: str -class ClassifiedCall(Call): +class ClassifiedCallTrace(CallTrace): inputs: Dict[str, Any] abi_name: str diff --git a/tests/helpers.py b/tests/helpers.py index 81b50d2..d27f57e 100644 --- a/tests/helpers.py +++ b/tests/helpers.py @@ -4,8 +4,8 @@ from mev_inspect.schemas.blocks import TraceType from mev_inspect.schemas.classified_traces import ( Classification, ClassifiedTrace, - Call, - ClassifiedCall, + CallTrace, + ClassifiedCallTrace, ) @@ -18,7 +18,7 @@ def make_transfer_trace( token_address: str, amount: int, ): - return Call( + return CallTrace( transaction_hash=transaction_hash, block_number=block_number, type=TraceType.call, @@ -43,7 +43,7 @@ def make_swap_trace( recipient_address: str, recipient_input_key: str, ): - return ClassifiedCall( + return ClassifiedCallTrace( transaction_hash=transaction_hash, block_number=block_number, type=TraceType.call,